Searched refs:num_devices_per_task (Results 1 – 8 of 8) sorted by relevance
453 gp->num_devices_per_task.clear(); in SetDevPerTask()460 gp->num_devices_per_task[*last_task_name] = count; in SetDevPerTask()465 gp->num_devices_per_task[*last_task_name] = count; in SetDevPerTask()469 for (const auto& task_dev : gp->num_devices_per_task) { in SetDevPerTask()
462 col_params.group.num_devices_per_task.at(task_name); in UnblockDependencies()
248 col_params_->group.num_devices_per_task[task_name] = num_devices; in Init()
74 const int num_local_devices = col_params->group.num_devices_per_task.at( in Enqueue()
259 int num_devices_per_task) { in MakeDeviceSet() argument262 devices.reserve(num_tasks * num_devices_per_task + num_tasks + 1); in MakeDeviceSet()272 for (int device = 0; device < num_devices_per_task; ++device) in MakeDeviceSet()
69 for (const auto& dpt : num_devices_per_task) { in ToString()
73 std::unordered_map<string, int32> num_devices_per_task; member
142 col_params_->group.num_devices_per_task[task_name] = num_ranks; in Init()