[0-size Tensor Job2 No.36] Add 0-size Tensor support for index_add

[co63oc]

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

bool IndexAddOpInferSymbolicShape(

out 维度和x 维度相同，反向时 value填充0，torch不支持x, value不同shape，反向时value为0-size

import torch
x = torch.randn([3, 0])
value = torch.randn([3, 1])
x.requires_grad=True
value.requires_grad=True
out = torch.index_add(x, 0, torch.randn([3]).type(torch.int32), value)
out.sum().backward()
print(x.grad)
print(value.grad)

RuntimeError: source tensor shape must match self tensor shape, excluding the specified dimension. Got self.shape = [3, 0] source.shape = [3, 1]

PaddleAPITest中有index 为0-size测试，但是torch index_add并不支持

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

PaddleAPITest 使用的是其他组合计算

paddle组合实现代码，当前代码没有支持value的反向，没有增加index 为0-size 的单测

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

PaddleAPITest测试通过，错误为numpy error

[paddle-bot]

你的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.

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 0% with 5 lines in your changes missing coverage. Please review.

Please upload report for BASE (develop@3c5a95c). Learn more about missing BASE report.

Files with missing lines	Patch %	Lines
paddle/phi/kernels/cpu/index_add_grad_kernel.cc	0.00%	5 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             develop   #73517   +/-   ##
==========================================
  Coverage           ?    0.00%           
==========================================
  Files              ?        1           
  Lines              ?        5           
  Branches           ?        0           
==========================================
  Hits               ?        0           
  Misses             ?        5           
  Partials           ?        0           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[co63oc]

/re-run all-failed

[DanielSun11]

LGTM