From 720af40aee438117562b08f232d1c4ca56d1e32c Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Tue, 25 Oct 2022 14:11:48 -0700 Subject: [PATCH] Test ResNet on MPS After https://github.com/pytorch/pytorch/issues/86954 is fixed, we should be able to test resnet on MPS --- test/smoke_test/smoke_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index 0ddcd267a..be8f25c90 100644 --- 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()