OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[CI] Fix GPUTests.test_scheduler_vertical_fusion1 (#151166)

Browse Source 
By enabling the test_operators on MPS device

Pull Request resolved: https://github.com/pytorch/pytorch/pull/151166
Approved by: https://github.com/dcci
This commit is contained in:
Nikita Shulga 2025-04-12 13:24:08 -07:00 committed by PyTorch MergeBot
parent 9699cc3eb9
commit d289d1177c
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/inductor/test_mps_basic.py
View File

@ -223,6 +223,7 @@ for test_name in [
"test_rsqrt", "test_rsqrt",
"test_scalar_cpu_tensor_arg", "test_scalar_cpu_tensor_arg",
"test_scalar_output", "test_scalar_output",
"test_scheduler_vertical_fusion1",
"test_setitem_with_int_parameter", "test_setitem_with_int_parameter",
"test_signbit", "test_signbit",
"test_silu", "test_silu",

2
torch/_inductor/test_operators.py
View File

@ -12,7 +12,7 @@ if not torch._running_with_deploy():
) )
_test_lib_impl = torch.library.Library("_inductor_test", "IMPL") _test_lib_impl = torch.library.Library("_inductor_test", "IMPL")
for dispatch_key in ("CPU", "CUDA", "Meta"): for dispatch_key in ("CPU", "CUDA", "MPS", "Meta"):
_test_lib_impl.impl("realize", lambda x: x.clone(), dispatch_key) _test_lib_impl.impl("realize", lambda x: x.clone(), dispatch_key)
class Realize(Function): class Realize(Function):