Skip to content

Commit

Permalink
Fix typo in t2 error test
Browse files Browse the repository at this point in the history
  • Loading branch information
WardBrian committed Oct 13, 2023
1 parent f47f0e1 commit fcc2072
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_errors.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit fcc2072

Please sign in to comment.