Skip to content

GitLab

Unverified Commit feda8b7b authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Extend TorchVision smoke tests to MPS (#7365)

parent 120e7af6
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
test/smoke_test.py
View file @ feda8b7b
......@@ -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__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment