From e6ebbca28ef7b43ce3e5bcb091be37b6b2f0cdcf Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Wed, 26 Oct 2022 14:53:54 -0700 Subject: [PATCH] Revert "Test ResNet on MPS (#1176)" This reverts commit efa1bc745bfc1dd4a2adc92fb4fdfe19dd745d4c. --- test/smoke_test/smoke_test.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py index be8f25c90..0ddcd267a 100644 --- a/test/smoke_test/smoke_test.py +++ b/test/smoke_test/smoke_test.py @@ -208,8 +208,6 @@ 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()