Adjust half-type threshold for cross product unit tests (#838)

* Adjust half-type threshold for cross product unit tests * Use the same seed for rng each time
NVIDIA · Jan 23, 2025 · d6a9c5c · d6a9c5c
1 parent 30799b3
commit d6a9c5c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/00_operators/OperatorTests.cu b/test/00_operators/OperatorTests.cu
@@ -2015,7 +2015,7 @@ TYPED_TEST(OperatorTestsFloatNonComplexAllExecs, Cross)
   // Half precision needs a bit more tolerance when compared to fp32
   float thresh = 0.01f;
   if constexpr (is_matx_half_v<TestType>) {
-    thresh = 0.02f;
+    thresh = 0.08f;
   }
 
   {//batched 4 x 3

diff --git a/test/test_vectors/generators/00_operators.py b/test/test_vectors/generators/00_operators.py
@@ -24,7 +24,8 @@ def run(self) -> Dict[str, np.array]:
 class cross_operator:
     def __init__(self, dtype: str, size: List[int]):
         self.size = size
-        self.dtype = dtype        
+        self.dtype = dtype
+        np.random.seed(1234)        
         pass
 
     def run(self) -> Dict[str, np.array]: