(inputs, device_ids)
| 67 | |
| 68 | |
| 69 | def _async_copy(inputs, device_ids): |
| 70 | nr_devs = len(device_ids) |
| 71 | assert type(inputs) in (tuple, list) |
| 72 | assert len(inputs) == nr_devs |
| 73 | |
| 74 | outputs = [] |
| 75 | for i, dev in zip(inputs, device_ids): |
| 76 | with cuda.device(dev): |
| 77 | outputs.append(async_copy_to(i, dev)) |
| 78 | |
| 79 | return tuple(outputs) |
| 80 | |
| 81 | |
| 82 | def _async_copy_stream(inputs, device_ids): |
nothing calls this directly
no test coverage detected