Fix vector type

NetFabric · Feb 27, 2024 · 1fcf3c4 · 1fcf3c4
1 parent f3ab532
commit 1fcf3c4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/src/NetFabric.Numerics.Tensors/IndexOfPredicateBinary.cs b/src/NetFabric.Numerics.Tensors/IndexOfPredicateBinary.cs
@@ -22,15 +22,15 @@ public static int IndexOfPredicate<T, TPredicateOperator>(ReadOnlySpan<T> x, T y
                 ref var currentVector = ref Unsafe.Add(ref vectorsRef, indexVector);
                 if (TPredicateOperator.Invoke(ref currentVector, ref yVector))
                 {
-                    for (var indexElement = 0; indexElement < Vector<int>.Count; indexElement++)
+                    for (var indexElement = 0; indexElement < Vector<T>.Count; indexElement++)
                     {
                         if (TPredicateOperator.Invoke(currentVector[indexElement], y))
-                            return ((int)indexVector * Vector<int>.Count) + indexElement;
+                            return ((int)indexVector * Vector<T>.Count) + indexElement;
                     }
                 }
             }
 
-            indexSource = indexVector * Vector<int>.Count;
+            indexSource = indexVector * Vector<T>.Count;
         }
 
         ref var xRef = ref MemoryMarshal.GetReference(x);

diff --git a/src/NetFabric.Numerics.Tensors/IndexOfPredicateTernary.cs b/src/NetFabric.Numerics.Tensors/IndexOfPredicateTernary.cs
@@ -23,15 +23,15 @@ public static int IndexOfPredicate<T, TPredicateOperator>(ReadOnlySpan<T> x, T y
                 ref var currentVector = ref Unsafe.Add(ref vectorsRef, indexVector);
                 if (TPredicateOperator.Invoke(ref currentVector, ref yVector, ref zVector))
                 {
-                    for (var indexElement = 0; indexElement < Vector<int>.Count; indexElement++)
+                    for (var indexElement = 0; indexElement < Vector<T>.Count; indexElement++)
                     {
                         if (TPredicateOperator.Invoke(currentVector[indexElement], y, z))
-                            return ((int)indexVector * Vector<int>.Count) + indexElement;
+                            return ((int)indexVector * Vector<T>.Count) + indexElement;
                     }
                 }
             }
 
-            indexSource = indexVector * Vector<int>.Count;
+            indexSource = indexVector * Vector<T>.Count;
         }
 
         ref var xRef = ref MemoryMarshal.GetReference(x);

diff --git a/src/NetFabric.Numerics.Tensors/IndexOfPredicateUnary.cs b/src/NetFabric.Numerics.Tensors/IndexOfPredicateUnary.cs
@@ -21,15 +21,15 @@ public static int IndexOfPredicate<T, TPredicateOperator>(ReadOnlySpan<T> x)
                 ref var currentVector = ref Unsafe.Add(ref vectorsRef, indexVector);
                 if (TPredicateOperator.Invoke(ref currentVector))
                 {
-                    for (var indexElement = 0; indexElement < Vector<int>.Count; indexElement++)
+                    for (var indexElement = 0; indexElement < Vector<T>.Count; indexElement++)
                     {
                         if (TPredicateOperator.Invoke(currentVector[indexElement]))
-                            return ((int)indexVector * Vector<int>.Count) + indexElement;
+                            return ((int)indexVector * Vector<T>.Count) + indexElement;
                     }
                 }
             }
 
-            indexSource = indexVector * Vector<int>.Count;
+            indexSource = indexVector * Vector<T>.Count;
         }
 
         ref var xRef = ref MemoryMarshal.GetReference(x);