Fix device-aware dtype handling in identity, gradient functions #8944
| Job | Run time |
|---|---|
| 24m 51s | |
| 24m 38s | |
| 18m 33s | |
| 20m 49s | |
| 24m 34s | |
| 19m 16s | |
| 20m 5s | |
| 18m 15s | |
| 19m 21s | |
| 25m 3s | |
| 1h 9m 36s | |
| 12m 26s | |
| 14m 13s | |
| 13m 49s | |
| 12m 31s | |
| 14m 18s | |
| 4m 15s | |
| 13m 52s | |
| 25m 3s | |
| 11m 59s | |
| 11m 56s | |
| -1s | |
| -2s | |
| 6h 59m 20s |