Add support for MPS
Backend [without torch.amp.autocast ] + CI (#3041)
#4469
Job | Run time |
---|---|
12m 46s | |
2m 38s | |
2m 30s | |
17m 54s |
MPS
Backend [without torch.amp.autocast ] + CI (#3041)
#4469
Job | Run time |
---|---|
12m 46s | |
2m 38s | |
2m 30s | |
17m 54s |