Extend TorchVision smoke tests to MPS (#7365)

pytorch · Mar 1, 2023 · feda8b7 · feda8b7
1 parent 120e7af
commit feda8b7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/smoke_test.py b/test/smoke_test.py
@@ -59,6 +59,8 @@ def main() -> None:
     smoke_test_torchvision_resnet50_classify()
     if torch.cuda.is_available():
         smoke_test_torchvision_resnet50_classify("cuda")
+    if torch.backends.mps.is_available():
+        smoke_test_torchvision_resnet50_classify("mps")
 
 
 if __name__ == "__main__":