Skip to content

Commit efa1bc7

Browse files
authored
Test ResNet on MPS (#1176)
After pytorch/pytorch#86954 is fixed, we should be able to test resnet on MPS
1 parent 95dbc19 commit efa1bc7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/smoke_test/smoke_test.py

Lines changed: 2 additions & 0 deletions
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)