update (#63324)

Author: feifei-111

paddle/cinn/hlir/dialect/operator/transforms/add_store_in_fusion_op_pass.cc

@@ -39,6 +39,7 @@ class AddYieldStoreInFusionOpPattern
         continue;
       }
 
+      rewriter.SetInsertionPointAfter(op->operand_source(i).defining_op());
       auto store_op = rewriter.Build<cinn::dialect::YieldStoreOp>(
           op->operand_source(i), op->operand_source(i).type());
       auto orignal_base = op->operand_source(i);

test/ir/pir/cinn/sub_graphs/test_sub_graph_47.py

@@ -63,7 +63,7 @@ def train(self, net, to_static, with_prim=False, with_cinn=False):
     def test_ast_prim_cinn(self):
         st_out = self.train(self.net, to_static=True)
         cinn_out = self.train(
-            self.net, to_static=True, with_prim=True, with_cinn=False
+            self.net, to_static=True, with_prim=True, with_cinn=True
         )
         for st, cinn in zip(
             paddle.utils.flatten(st_out), paddle.utils.flatten(cinn_out)

test/ir/pir/cinn/sub_graphs/test_sub_graph_77.py

@@ -174,7 +174,7 @@ def train(self, net, to_static, with_prim=False, with_cinn=False):
     def test_ast_prim_cinn(self):
         st_out = self.train(self.net, to_static=True)
         cinn_out = self.train(
-            self.net, to_static=True, with_prim=True, with_cinn=False
+            self.net, to_static=True, with_prim=True, with_cinn=True
         )
         for st, cinn in zip(
             paddle.utils.flatten(st_out), paddle.utils.flatten(cinn_out)