您当前的位置: 首页 > 

把几个任务分配到几个设备上的代码

柳鲲鹏 发布时间:2019-03-11 11:15:23 ,浏览量:0

  比如,吾有几个任务,想平均分配到几个设备上,怎么办?

  1. 如果能均分当然好办。
  2. 不能均分,把多出的任务分给前几个。
//先计算每个GPU上要承载几个任务

//平均分配
int tpg    = task_count/nGpuCount;

//如果有剩下的,再加1。这样前几个GPU多一个任务。
int left   = task_count % nGpuCount;
if (left != 0)
{
    tpg++;
}

//以任务序号,计算使用哪个GPU
int gpu_id = i/tpg;

 

关注
打赏
1688896170
查看更多评论

柳鲲鹏

暂无认证

  • 0浏览

    0关注

    4641博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文
立即登录/注册

微信扫码登录

0.0433s