比如,吾有几个任务,想平均分配到几个设备上,怎么办?
如果能均分当然好办。
不能均分,把多出的任务分给前几个。
//先计算每个GPU上要承载几个任务 //平均分配 int tpg = task_count/nGpuCount; //如果有剩下的,再加1。这样前几个GPU多一个任务。 int left = task_count % nGpuCount; if (left != 0) { tpg++; } //以任务序号,计算使用哪个GPU int gpu_id = i/tpg;
比如,吾有几个任务,想平均分配到几个设备上,怎么办?
如果能均分当然好办。
不能均分,把多出的任务分给前几个。
//先计算每个GPU上要承载几个任务 //平均分配 int tpg = task_count/nGpuCount; //如果有剩下的,再加1。这样前几个GPU多一个任务。 int left = task_count % nGpuCount; if (left != 0) { tpg++; } //以任务序号,计算使用哪个GPU int gpu_id = i/tpg;