Skip to content

Commit 32dc127

Browse files
authored
Merge branch 'main' into frost/nms_xpu_only
2 parents 7c974a5 + 4aca477 commit 32dc127

File tree

1 file changed

+0
-17
lines changed

1 file changed

+0
-17
lines changed

test/regressions/test_torchvision_roi_ops.py

-17
Original file line numberDiff line numberDiff line change
@@ -471,7 +471,6 @@ def expected_fn(
471471
) # , ids=str)
472472
@pytest.mark.parametrize("contiguous", (True, False))
473473
@pytest.mark.parametrize("deterministic", (True, False))
474-
@pytest.mark.opcheck_only_one
475474
def test_forward(
476475
self, device, contiguous, deterministic, aligned, x_dtype, rois_dtype=None
477476
):
@@ -486,22 +485,6 @@ def test_forward(
486485
aligned=aligned,
487486
)
488487

489-
@pytest.mark.parametrize("aligned", (True, False))
490-
@pytest.mark.parametrize("deterministic", (True, False))
491-
@pytest.mark.parametrize("x_dtype", (torch.float, torch.half))
492-
@pytest.mark.parametrize("rois_dtype", (torch.float, torch.half))
493-
@pytest.mark.opcheck_only_one
494-
def test_autocast(self, aligned, deterministic, x_dtype, rois_dtype):
495-
with torch.amp.autocast("xpu"):
496-
self.test_forward(
497-
torch.device("xpu"),
498-
contiguous=False,
499-
deterministic=deterministic,
500-
aligned=aligned,
501-
x_dtype=x_dtype,
502-
rois_dtype=rois_dtype,
503-
)
504-
505488

506489
class TestPSRoIAlign(RoIOpTester):
507490
mps_backward_atol = 5e-2

0 commit comments

Comments
 (0)