From fcc207262eb0e693c3e0c67871498aaed1bf2f14 Mon Sep 17 00:00:00 2001 From: Brian Ward Date: Fri, 13 Oct 2023 13:10:13 -0400 Subject: [PATCH] Fix typo in t2 error test --- tests/test_errors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_errors.py b/tests/test_errors.py index 0386ab5..93421fb 100644 --- a/tests/test_errors.py +++ b/tests/test_errors.py @@ -33,7 +33,7 @@ def test_t2_mismatch_device_cuda_cpu() -> None: targets = torch.randn(*g[0].shape, dtype=torch.complex128).to("cuda:0") with pytest.raises(ValueError, match="Some tensors are not on the same device"): - pytorch_finufft.functional.finufft_type1.apply(points, targets) + pytorch_finufft.functional.finufft_type2.apply(points, targets) @pytest.mark.skipif(torch.cuda.device_count() < 2, reason="require multiple GPUs")