Skip to content

Commit 7bd9632

Browse files
NicolasHugpmeier
andauthored
[Cherry-pick for 0.15] Revert CP of " Extend TorchVision smoke tests to MPS (#7365) " (#7368)
Co-authored-by: Philip Meier <[email protected]>
1 parent 6eb3efd commit 7bd9632

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

test/smoke_test.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,6 @@ def main() -> None:
5959
smoke_test_torchvision_resnet50_classify()
6060
if torch.cuda.is_available():
6161
smoke_test_torchvision_resnet50_classify("cuda")
62-
if torch.backends.mps.is_available():
63-
smoke_test_torchvision_resnet50_classify("mps")
6462

6563

6664
if __name__ == "__main__":

0 commit comments

Comments
 (0)