mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[MPS] Handle MPS failures of test_modules.py in common_modules.py (#95334)
- Also cleaned up `test_modules.py` from skipMPS code. - Added `skipMPS` for unsupported or failing tests on MPS backend in common_modules.py. (We'll remove `skipMPS` from those tests once a fix is available for them.) Pull Request resolved: https://github.com/pytorch/pytorch/pull/95334 Approved by: https://github.com/kulinseth, https://github.com/albanD
This commit is contained in:
committed by
PyTorch MergeBot
parent
97bb4c2538
commit
cecfcf1e17
@ -1265,7 +1265,7 @@ def get_selected_tests(options) -> List[ShardedTest]:
|
||||
options.exclude.extend(CPP_TESTS)
|
||||
|
||||
if options.mps:
|
||||
selected_tests = ["test_mps", "test_metal"]
|
||||
selected_tests = ["test_mps", "test_metal", "test_modules"]
|
||||
else:
|
||||
# Exclude all mps tests otherwise
|
||||
options.exclude.extend(["test_mps", "test_metal"])
|
||||
|
||||
Reference in New Issue
Block a user