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__":