fixed Gru

Luohongzhige · Luohongzhige · commit c4dd4b1e5bc9 · 2024-08-26T22:08:30.000+08:00
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/multiary_infer_sym.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/multiary_infer_sym.cc
@@ -1400,8 +1400,9 @@ bool GruOpInferSymbolicShape(pir::Operation *op,
     infer_context->SetShapeOrDataForValue(op->result(2), batch_hidden_shape);
   }
 
-  symbol::TensorShapeOrDataDimExprs hidden_shape({input_shape[0], frame_size});
-  infer_context->SetShapeOrDataForValue(op->result(3), hidden_shape);
+  symbol::TensorShapeOrDataDimExprs hidden_shape_output(
+      {input_shape[0], frame_size});
+  infer_context->SetShapeOrDataForValue(op->result(3), hidden_shape_output);
 
   return true;
 }

Original file line number	Diff line number	Diff line change
`@@ -1400,8 +1400,9 @@ bool GruOpInferSymbolicShape(pir::Operation *op,`
`1400`	`1400`	`infer_context->SetShapeOrDataForValue(op->result(2), batch_hidden_shape);`
`1401`	`1401`	`}`
`1402`	`1402`
`1403`		`- symbol::TensorShapeOrDataDimExprs hidden_shape({input_shape[0], frame_size});`
`1404`		`- infer_context->SetShapeOrDataForValue(op->result(3), hidden_shape);`
	`1403`	`+ symbol::TensorShapeOrDataDimExprs hidden_shape_output(`
	`1404`	`+ {input_shape[0], frame_size});`
	`1405`	`+ infer_context->SetShapeOrDataForValue(op->result(3), hidden_shape_output);`
`1405`	`1406`
`1406`	`1407`	`return true;`
`1407`	`1408`	`}`