We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 95dbc19 commit efa1bc7Copy full SHA for efa1bc7
test/smoke_test/smoke_test.py
@@ -208,6 +208,8 @@ def main() -> None:
208
smoke_test_torchvision_resnet50_classify()
209
if torch.cuda.is_available():
210
smoke_test_torchvision_resnet50_classify("cuda")
211
+ if torch.backends.mps.is_available():
212
+ smoke_test_torchvision_resnet50_classify("mps")
213
214
if __name__ == "__main__":
215
main()
0 commit comments