You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import torch
torch.index_add(torch.randn([101, 3]), 0, torch.randn([0]).type(torch.int32), torch.randn([3, 3]))
RuntimeError: index_add_(): Number of indices (0) should be equal to source.size(dim): (3), for dim: 0
import torch
x = torch.randn([10, 3])
index = torch.randn([0]).type(torch.int64)
axis = 0
value = torch.randn([10, 3])
x.requires_grad = True
value.requires_grad = True
for i in range(len(index)):
if index[i].item() >= x.size(axis):
continue
tmp = x.select(dim=axis, index=index[i].item())
tmp += value.select(dim=axis, index=i)
result = x
result.sum().backward()
print(value.grad)
None
你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.
co63oc
changed the title
CI测试不review index_add
[0-size Tensor No.36] Add 0-size Tensor support for index_add
Jun 22, 2025
co63oc
changed the title
[0-size Tensor No.36] Add 0-size Tensor support for index_add
[0-size Tensor Job2 No.36] Add 0-size Tensor support for index_add
Jun 22, 2025
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
PR Category
Execute Infrastructure
PR Types
Improvements
Description
index_add
paddle/phi/infermeta/binary.cc 取消检查,symbolic 推导没有对应代码
Paddle/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc
Line 2614 in 0c29fe4
out 维度和x 维度相同,反向时 value填充0,torch不支持x, value不同shape,反向时value为0-size
PaddleAPITest中有index 为0-size测试,但是torch index_add并不支持

PaddleAPITest 使用的是其他组合计算

paddle组合实现代码,当前代码没有支持value的反向,没有增加index 为0-size 的单测
PaddleAPITest测试通过,错误为numpy error
