Returns a queue with [0, 1, .. num_gpus].
(num_gpus: int, manager: SyncManager)
| 268 | |
| 269 | |
| 270 | def _setup_gpu_queue(num_gpus: int, manager: SyncManager): |
| 271 | """Returns a queue with [0, 1, .. |
| 272 | |
| 273 | num_gpus]. |
| 274 | """ |
| 275 | gpu_queue = manager.Queue(num_gpus) |
| 276 | for gpu_id in range(num_gpus): |
| 277 | gpu_queue.put(gpu_id) |
| 278 | return gpu_queue |
| 279 | |
| 280 | |
| 281 | def create_job_configs( |