diff --git a/test/xpu/xpu_test_utils.py b/test/xpu/xpu_test_utils.py index 4b0900e967..80b01bac00 100644 --- a/test/xpu/xpu_test_utils.py +++ b/test/xpu/xpu_test_utils.py @@ -353,6 +353,10 @@ ("_refs.mul", "test_python_ref"), ("_refs.mul", "test_python_ref_torch_fallback"), ("nn.AvgPool2d", "test_memory_format"), + ("nn.LazyConv2d", "test_memory_format"), + ("nn.Conv2d", "test_memory_format"), + ("nn.ConvTranspose2d", "test_memory_format"), + ("nn.LazyConvTranspose2d", "test_memory_format"), ("narrow_copy", "test_meta_outplace"), ("narrow_copy", "test_dispatch_meta_outplace"), ("narrow_copy", "test_dispatch_symbolic_meta_outplace"), @@ -361,6 +365,7 @@ ("_refs.pow", "test_python_ref"), ("_refs.pow", "test_python_ref_torch_fallback"), ("_refs.mul", "test_python_ref_executor"), + ("_refs.pow", "test_python_ref_executor"), ( "_refs.div", "test_python_ref_torch_fallback", @@ -1016,6 +1021,13 @@ def gen_xpu_wrappers(op_name, wrappers): else: wrapper.device_type = "xpu" replaced = True + elif ( + wrapper.device_type is None + and unittest.expectedFailure in wrapper.decorators + and (op_name, wrapper.test_name) in _cuda_xfail_xpu_pass + ): + replaced = True + continue wrapper_xpu.append(wrapper) elif self.only_cuda_fn == wrapper: wrapper_xpu.append(common_device_type.onlyCUDA)