[Backport] Move CUDA tensors to CPU before moving to XLA #3474
Job | Run time |
---|---|
45m 50s | |
0s | |
0s | |
0s | |
13m 13s | |
19m 20s | |
14m 2s | |
26m 41s | |
17m 39s | |
1h 22m 56s | |
9m 54s | |
51m 11s | |
10m 53s | |
52m 11s | |
14m 53s | |
44m 13s | |
6h 42m 56s |
Job | Run time |
---|---|
45m 50s | |
0s | |
0s | |
0s | |
13m 13s | |
19m 20s | |
14m 2s | |
26m 41s | |
17m 39s | |
1h 22m 56s | |
9m 54s | |
51m 11s | |
10m 53s | |
52m 11s | |
14m 53s | |
44m 13s | |
6h 42m 56s |