frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix XPU CI][Inductor UT] Fix test cases broken by community. (#160403)

Browse Source 
Fixes #160243, Fixes #160244, Fixes #160245

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160403
Approved by: https://github.com/janeyx99
This commit is contained in:
xinan.lin
2025-08-15 23:33:25 -07:00
committed by PyTorch MergeBot
parent bbc7c03e93
commit 1853f71b4f
4 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/inductor/test_torchinductor_codegen_dynamic_shapes.py
View File

@ -348,7 +348,7 @@ test_failures = {
"test_rand_like_deterministic_dynamic_shapes": TestFailure(
("cpu", "cuda", "xpu"), is_skip=True
),
"test_repeat_interleave_2_dynamic_shapes": TestFailure(("cpu", "xpu")),
"test_repeat_interleave_2_dynamic_shapes": TestFailure(("cpu",)),
"test_slice_mutation2_dynamic_shapes": TestFailure(
("cpu", "cuda", "xpu"), is_skip=True
),

8
test/inductor/test_torchinductor_opinfo.py
View File

@ -682,6 +682,14 @@ inductor_override_kwargs["xpu"] = {
("nn.functional.unfold", f16): {
"reference_in_float": True,
},
# Reference crash on Intel LTS2 driver.
("nn.functional.interpolate.trilinear", f32): {
"check_gradient": False,
},
# Reference crash on Intel LTS2 driver.
("nn.functional.interpolate.trilinear", f64): {
"check_gradient": False,
},
}
if TEST_WITH_ROCM:
inductor_override_kwargs["cuda"].update(

2
test/test_testing.py
View File

@ -2351,7 +2351,7 @@ class TestImports(TestCase):
# fail, so just set CWD to this script's directory
cwd=os.path.dirname(os.path.realpath(__file__)),).decode("utf-8")
# The test is flaky on ROCm and has been open and close multiple times
# The test is flaky on ROCm/XPU and has been open and close multiple times
# https://github.com/pytorch/pytorch/issues/110040
@skipIfRocm
def test_circular_dependencies(self) -> None:

4
torch/_inductor/ir.py
View File

@ -7160,10 +7160,10 @@ class DeviceCopy(ExternKernelOut):
# x.get_stride() may be unimplemented if x's size is empty
stride = x.get_stride()
is_destination_pinned = (
x_device.type == "cuda" and device.type == "cpu" and non_blocking
is_gpu(x_device.type) and device.type == "cpu" and non_blocking
)
is_source_pinned = (
x_device.type == "cpu" and device.type == "cuda" and non_blocking
x_device.type == "cpu" and is_gpu(device.type) and non_blocking
)
if is_source_pinned and is_storage_and_layout(x):
x.get_layout().is_pinned = True