Add support for MPS Backend [without torch.amp.autocast ] + CI (#3041)
#4820
| Job | Run time |
|---|---|
| 17m 29s | |
| 20m 28s | |
| 18m 11s | |
| 21m 15s | |
| 45m 11s | |
| 21m 30s | |
| 17m 10s | |
| 21m 5s | |
| 12m 1s | |
| 11m 55s | |
| 3h 26m 15s |