[co63oc]

PR Category

Execute Infrastructure

PR Types

Improvements

Description

paddle.take 调用index_select
修改index_select infermeta检查
IndexSelectOpInferSymbolicShape 没有对应代码

Paddle/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/binary_infer_sym.cc

Line 2514 in d24f6a4

bool IndexSelectOpInferSymbolicShape(

math.py max_index 需要判断大于0

修改kernel为 index_select

PaddleAPITest 测试通过

high < 0 是PaddleAPITest中randint 调用时，high为0，如果输入为0-size，但是index不为0-size产生，torch 中同样会有错误

import torch
torch.take(torch.randn([3,0]), torch.randn([2, 3]).type(torch.int64))

[codecov-commenter]

Codecov Report

❌ Patch coverage is 25.00000% with 6 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (develop@fa0f36f). Learn more about missing BASE report.

Files with missing lines	Patch %	Lines
paddle/phi/kernels/cpu/index_select_grad_kernel.cc	0.00%	3 Missing ⚠️
paddle/phi/kernels/cpu/index_select_kernel.cc	0.00%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             develop   #73186   +/-   ##
==========================================
  Coverage           ?   25.00%           
==========================================
  Files              ?        3           
  Lines              ?        8           
  Branches           ?        0           
==========================================
  Hits               ?        2           
  Misses             ?        6           
  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.

[DanielSun11]

LGTM