【PIR API adaptor No.193-196】Migrate paddle.geometric.segment_max, paddle.geometric.segment_mean, paddle.geometric.segment_min, paddle.geometric.segment_sum into pir

[enkilee]

PR types

Others

PR changes

APIs

Description

• 新IR Python API适配升级 #58067

PIR API 推全升级
将 paddle.geometric.segment_max 迁移升级至 pir，并更新单测 单测覆盖率：5/6
将 paddle.geometric.segment_mean 迁移升级至 pir，并更新单测 单测覆盖率：5/6
将 paddle.geometric.segment_min 迁移升级至 pir，并更新单测 单测覆盖率：5/6
将 paddle.geometric.segment_sum 迁移升级至 pir，并更新单测 单测覆盖率：5/6

check_grad 加上 check_pir=True报错：

2023-11-01 19:44:21 --------------------------------------
2023-11-01 19:44:21 C++ Traceback (most recent call last):
2023-11-01 19:44:21 --------------------------------------
2023-11-01 19:44:21 0   paddle::framework::ThreadPoolTempl<paddle::framework::StlThreadEnvironment>::WorkerLoop(int)
2023-11-01 19:44:21 1   paddle::framework::PirInterpreter::RunInstructionBaseAsync(unsigned long)
2023-11-01 19:44:21 2   paddle::framework::PirInterpreter::RunInstructionBase(paddle::framework::InstructionBase*)
2023-11-01 19:44:21 3   paddle::framework::PhiKernelInstruction::Run()
2023-11-01 19:44:21 4   phi::KernelImpl<void (*)(phi::CPUContext const&, phi::DenseTensor const&, paddle::experimental::ScalarBase<phi::DenseTensor> const&, phi::DataType, phi::DenseTensor*), &(void phi::FullLikeKernel<double, phi::CPUContext>(phi::CPUContext const&, phi::DenseTensor const&, paddle::experimental::ScalarBase<phi::DenseTensor> const&, phi::DataType, phi::DenseTensor*))>::Compute(phi::KernelContext*)
2023-11-01 19:44:21 5   void phi::FullValue<double, phi::CPUContext, double>(phi::CPUContext const&, phi::DenseTensor*, double)
2023-11-01 19:44:21 6   phi::enforce::EnforceNotMet::EnforceNotMet(phi::ErrorSummary const&, char const*, int)
2023-11-01 19:44:21 7   phi::enforce::GetCurrentTraceBackString[abi:cxx11](bool)
2023-11-01 19:44:21 
2023-11-01 19:44:21 ----------------------
2023-11-01 19:44:21 Error Message Summary:
2023-11-01 19:44:21 ----------------------
2023-11-01 19:44:21 PreconditionNotMetError: The meta data must be valid when call the mutable data function.
2023-11-01 19:44:21   [Hint: Expected valid() == true, but received valid():0 != true:1.] (at ../paddle/phi/core/dense_tensor.cc:123)
2023-11-01 19:44:21   [operator < pd_kernel.phi_kernel > error]

[0x45f]

ERROR: test_static (test_segment_ops.API_SegmentOpsTest) 报错是因为这个测试case中调用的是 paddle.incubate.segment_sum API，但是本PR中适配的是 paddle.geometric.segment_sum API。辛苦在这个PR中将这两个API都适配一下新IR吧，然后在尝试打开这个单测case~

[enkilee]

已改

[0x45f]

反向的问题我们内部来看下~

[xiaoguoguo626807]

#59036 修复反向报错