Test ResNet on MPS (pytorch#1176)

After pytorch/pytorch#86954 is fixed, we should be able to test resnet on MPS
ROCm · Oct 26, 2022 · efa1bc7 · efa1bc7
1 parent 95dbc19
commit efa1bc7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -208,6 +208,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__":
     main()