From b71473f0a3cb737ecee35fd1263ae84490e56b6d Mon Sep 17 00:00:00 2001 From: Brian Ward Date: Mon, 5 Feb 2024 11:24:15 -0500 Subject: [PATCH] Fix forward test tolerance --- tests/test_t3_forward.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_t3_forward.py b/tests/test_t3_forward.py index 9deb7df..0b2142c 100644 --- 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)