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()