Skip to content

Commit 2f20817

Browse files
authored
Merge branch 'main' into mengfeil/enhance-workflow
2 parents 2662b13 + d799bd3 commit 2f20817

File tree

2 files changed

+1
-18
lines changed

2 files changed

+1
-18
lines changed

CMakeLists.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ include(${TORCH_XPU_OPS_ROOT}/cmake/ONEMKL.cmake)
4040
include(${TORCH_XPU_OPS_ROOT}/cmake/BuildFlags.cmake)
4141

4242
option(USE_XCCL "Build with XCCL support" OFF)
43-
option(USE_C10D_XCCL "Build with XCCL support for C10D" OFF)
43+
option(USE_C10D_XCCL "Build with XCCL support for C10D" ON)
4444

4545
# -- [ Re-generate the macros file for https://github.com/pytorch/pytorch/pull/147161
4646
macro(update_caffe2_macros_file)

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)