Skip to content

Commit

Permalink
[Cherry-pick for 0.15] Revert CP of " Extend TorchVision smoke tests …
Browse files Browse the repository at this point in the history
…to MPS (#7365) " (#7368)

Co-authored-by: Philip Meier <[email protected]>
  • Loading branch information
NicolasHug and pmeier authored Mar 1, 2023
1 parent 6eb3efd commit 7bd9632
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions test/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,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__":
Expand Down

0 comments on commit 7bd9632

Please sign in to comment.