Add support for MPS Backend [without torch.amp.autocast ] + CI
#4819
| Job | Run time |
|---|---|
| 45m 12s | |
| 20m 28s | |
| 17m 5s | |
| 45m 14s | |
| 16m 35s | |
| 20m 57s | |
| 16m 53s | |
| 20m 14s | |
| 12m 26s | |
| 17m 56s | |
| 3h 53m 0s |