Fix forward test tolerance

flatironinstitute · Feb 5, 2024 · b71473f · b71473f
1 parent bd4d515
commit b71473f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_t3_forward.py b/tests/test_t3_forward.py
@@ -4,6 +4,8 @@
 
 import pytorch_finufft
 
+torch.manual_seed(45678)
+
 
 def check_t3_forward(N: int, dim: int, device: str) -> None:
     """
@@ -28,7 +30,7 @@ def check_t3_forward(N: int, dim: int, device: str) -> None:
     print("shape of targets is " + str(targets.shape))
 
     finufft_out = pytorch_finufft.functional.finufft_type3(
-        points, values.flatten(), targets
+        points, values.flatten(), targets, eps=1e-7
     )
 
     against_torch = torch.fft.fftn(values)