From 5ad04bdd42aa5c40a42433849392299e264be921 Mon Sep 17 00:00:00 2001 From: zhangbopd <1299246947@qq.com> Date: Tue, 9 Jan 2024 07:14:27 +0000 Subject: [PATCH] fix --- .../pir/dialect/operator/interface/infer_symbolic_shape.cc | 3 +++ paddle/fluid/pir/dialect/operator/ir/ops.yaml | 1 - 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape.cc index 478093595acee7..5d4cc10b205ba4 100644 --- a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape.cc +++ b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape.cc @@ -76,6 +76,9 @@ bool InferSymbolicShapeElementWiseBinary( pir::OpResult res = op->result(0); symbol::ShapeOrDataDimExprs shape_data{shapes, data}; shape_analysis->SetShapeOrDataForValue(res, shape_data); + op->set_attribute( + "symbolic_shape", + pir::shape::SymbolAttribute::get(pir::IrContext::Instance(), shape_data)); return true; } diff --git a/paddle/fluid/pir/dialect/operator/ir/ops.yaml b/paddle/fluid/pir/dialect/operator/ir/ops.yaml index 6f6f79ec8a5ed5..d9c5b1c8736099 100644 --- a/paddle/fluid/pir/dialect/operator/ir/ops.yaml +++ b/paddle/fluid/pir/dialect/operator/ir/ops.yaml @@ -1201,7 +1201,6 @@ func : softmax inplace : (x -> out) backward : softmax_grad - interfaces : paddle::dialect::InferSymbolicShapeInterface - op : split args : (Tensor x, IntArray sections, Scalar(int) axis)