feat: improve how device switch is handled between the metric device and the input tensors device #4703
| Job | Run time |
|---|---|
| 6m 6s | |
| 6m 4s | |
| 6m 7s | |
| 6m 5s | |
| 6m 5s | |
| 6m 8s | |
| 6m 6s | |
| 6m 6s | |
| 6m 1s | |
| 6m 0s | |
| 1h 0m 48s |
| Job | Run time |
|---|---|
| 6m 6s | |
| 6m 4s | |
| 6m 7s | |
| 6m 5s | |
| 6m 5s | |
| 6m 8s | |
| 6m 6s | |
| 6m 6s | |
| 6m 1s | |
| 6m 0s | |
| 1h 0m 48s |