https://gitcode.com/cann/ops-nn/blob/master/index/scatter_add/docs/aclnnScatterAdd.md
aclnnScatterAddGetWorkspaceSize 的参数说明表中,src 和 out 的“非连续Tensor”列均标为“×”。
aclnnScatterAddGetWorkspaceSize
src
out
这两个标注与现有接口实现不一致:
l0op::Contiguous
InitializeTensor
l0op::ViewCopy(scatterRes, out, executor)
实现依据:aclnn_scatter.cpp,其中 src 转连续位于第 364、589 行,NoTranspose 和通用路径的输出回写位于第 376、383、781 行。
aclnn_scatter.cpp
将 src 和 out 的“非连续Tensor”支持标记从“×”改为“√”,使文档准确反映现有行为。
文档链接
https://gitcode.com/cann/ops-nn/blob/master/index/scatter_add/docs/aclnnScatterAdd.md
问题文档片段
aclnnScatterAddGetWorkspaceSize的参数说明表中,src和out的“非连续Tensor”列均标为“×”。存在的问题
这两个标注与现有接口实现不一致:
src在 NoTranspose 路径通过l0op::Contiguous转连续;通用路径通过InitializeTensor调用同一转换后参与计算。l0op::ViewCopy(scatterRes, out, executor)回写调用者传入的原始out,适配其非连续视图。实现依据:
aclnn_scatter.cpp,其中src转连续位于第 364、589 行,NoTranspose 和通用路径的输出回写位于第 376、383、781 行。预期修正
将
src和out的“非连续Tensor”支持标记从“×”改为“√”,使文档准确反映现有行为。