modify assign inplace (#68751)

xiaoguoguo626807 · web-flow · commit af89954f972c · 2024-10-18T10:48:21.000+08:00
diff --git a/paddle/fluid/framework/new_executor/pir_adaptor/pir_adaptor_util.cc b/paddle/fluid/framework/new_executor/pir_adaptor/pir_adaptor_util.cc
@@ -692,7 +692,7 @@ bool IsNeedVarInplace(pir::Operation* op,
                       pir::Value value,
                       std::string op_name) {
   return (value.type().isa<paddle::dialect::DenseTensorArrayType>() ||
-          op_name == "pd_op.assign_value_");
+          op_name == "pd_op.assign_value_" || op_name == "pd_op.assign_out_");
 }
 
 // NOTE(chenxi67): Here, we only perform inplace processing for variables that

Original file line number	Diff line number	Diff line change
`@@ -692,7 +692,7 @@ bool IsNeedVarInplace(pir::Operation* op,`
`692`	`692`	`pir::Value value,`
`693`	`693`	`std::string op_name) {`
`694`	`694`	`return (value.type().isa<paddle::dialect::DenseTensorArrayType>() \|\|`
`695`		`- op_name == "pd_op.assign_value_");`
	`695`	`+ op_name == "pd_op.assign_value_" \|\| op_name == "pd_op.assign_out_");`
`696`	`696`	`}`
`697`	`697`
`698`	`698`	`// NOTE(chenxi67): Here, we only perform inplace processing for variables that`