Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit feda8b7

Browse files
authoredMar 1, 2023
Extend TorchVision smoke tests to MPS (#7365)
1 parent 120e7af commit feda8b7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed
 

‎test/smoke_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,8 @@ def main() -> None:
5959
smoke_test_torchvision_resnet50_classify()
6060
if torch.cuda.is_available():
6161
smoke_test_torchvision_resnet50_classify("cuda")
62+
if torch.backends.mps.is_available():
63+
smoke_test_torchvision_resnet50_classify("mps")
6264

6365

6466
if __name__ == "__main__":

0 commit comments

Comments
 (0)
Please sign in to comment.