华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
芯片类型:950
问题:GridSampler2DGrad算子对同一GM地址使用atomic_add写入数据时,等待耗时过长,导致算子性能下降。由于算子特殊性,无法预测写入地址,导致不能在ub上提前累加再往GM上写入。
测试:手动构造写入地址不同的用例,耗时15us;相同shape下,存在对同一GM地址写入数据的情况,耗时增加到64us,且相同地址越多,耗时越长。
我要发帖子
芯片类型:950
问题:GridSampler2DGrad算子对同一GM地址使用atomic_add写入数据时,等待耗时过长,导致算子性能下降。由于算子特殊性,无法预测写入地址,导致不能在ub上提前累加再往GM上写入。
测试:手动构造写入地址不同的用例,耗时15us;相同shape下,存在对同一GM地址写入数据的情况,耗时增加到64us,且相同地址越多,耗时越长。