From 4c027fee4c44b38e024aa86241ae62d2435052d4 Mon Sep 17 00:00:00 2001
From: Nikita Shulga <nshulga@fb.com>
Date: Tue, 28 Feb 2023 16:47:25 -0800
Subject: [PATCH] Extend TorchVision smoke tests to MPS

---
 test/smoke_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/smoke_test.py b/test/smoke_test.py
index 728c9440fd3..e8ee178d95e 100644
--- 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__":