Skip to content

Commit

Permalink
Merge branch 'main' into penghuic/update_matmul_cuda_cases
Browse files Browse the repository at this point in the history
  • Loading branch information
PenghuiCheng authored Mar 4, 2025
2 parents 5c76680 + 4aca477 commit d919c33
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions test/regressions/test_torchvision_roi_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -471,7 +471,6 @@ def expected_fn(
) # , ids=str)
@pytest.mark.parametrize("contiguous", (True, False))
@pytest.mark.parametrize("deterministic", (True, False))
@pytest.mark.opcheck_only_one
def test_forward(
self, device, contiguous, deterministic, aligned, x_dtype, rois_dtype=None
):
Expand All @@ -486,22 +485,6 @@ def test_forward(
aligned=aligned,
)

@pytest.mark.parametrize("aligned", (True, False))
@pytest.mark.parametrize("deterministic", (True, False))
@pytest.mark.parametrize("x_dtype", (torch.float, torch.half))
@pytest.mark.parametrize("rois_dtype", (torch.float, torch.half))
@pytest.mark.opcheck_only_one
def test_autocast(self, aligned, deterministic, x_dtype, rois_dtype):
with torch.amp.autocast("xpu"):
self.test_forward(
torch.device("xpu"),
contiguous=False,
deterministic=deterministic,
aligned=aligned,
x_dtype=x_dtype,
rois_dtype=rois_dtype,
)


class TestPSRoIAlign(RoIOpTester):
mps_backward_atol = 5e-2
Expand Down

0 comments on commit d919c33

Please sign in to comment.