Skip to content

Commit 720af40

Browse files
authored
Test ResNet on MPS
After pytorch/pytorch#86954 is fixed, we should be able to test resnet on MPS
1 parent 59140aa commit 720af40

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/smoke_test/smoke_test.py

+2
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,8 @@ def main() -> None:
208208
smoke_test_torchvision_resnet50_classify()
209209
if torch.cuda.is_available():
210210
smoke_test_torchvision_resnet50_classify("cuda")
211+
if torch.backends.mps.is_available():
212+
smoke_test_torchvision_resnet50_classify("mps")
211213

212214
if __name__ == "__main__":
213215
main()

0 commit comments

Comments
 (0)