fix

PaddlePaddle · Jan 9, 2024 · 5ad04bd · 5ad04bd
1 parent f7c6eed
commit 5ad04bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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
@@ -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)