Fix device-aware dtype handling in identity, gradient functions #8950
| Job | Run time |
|---|---|
| 18m 49s | |
| 19m 9s | |
| 19m 4s | |
| 19m 9s | |
| 24m 55s | |
| 25m 48s | |
| 25m 26s | |
| 24m 38s | |
| 23m 15s | |
| 19m 34s | |
| 14m 27s | |
| 12m 6s | |
| 13m 48s | |
| 13m 41s | |
| 12m 12s | |
| 12m 40s | |
| 13m 55s | |
| 12m 8s | |
| 39m 6s | |
| 26m 41s | |
| 4m 16s | |
| 0s | |
| 0s | |
| 6h 34m 47s |