SIMT中对GM上同一地址写入数据的写等待问题是否有解决方案
收藏回复举报
SIMT中对GM上同一地址写入数据的写等待问题是否有解决方案
t('forum.solved') 已解决
新人帖
发表于2026-05-30 11:44:08
0 查看

芯片类型:950

问题:GridSampler2DGrad算子对同一GM地址使用atomic_add写入数据时,等待耗时过长,导致算子性能下降。由于算子特殊性,无法预测写入地址,导致不能在ub上提前累加再往GM上写入。

测试:手动构造写入地址不同的用例,耗时15us;相同shape下,存在对同一GM地址写入数据的情况,耗时增加到64us,且相同地址越多,耗时越长。

我要发帖子