Add support for MPS
Backend [without torch.amp.autocast ] + CI (#3041)
#446
Job | Run time |
---|---|
27m 37s | |
31m 41s | |
59m 18s |
MPS
Backend [without torch.amp.autocast ] + CI (#3041)
#446
Job | Run time |
---|---|
27m 37s | |
31m 41s | |
59m 18s |