[Dy2St][CINN] Explicit use phi backend in more CINN uts (#71896)

SigureMo · web-flow · commit 2ac03aef5580 · 2025-03-26T12:49:12.000+08:00
diff --git a/test/ir/pir/cinn/sub_graphs/base.py b/test/ir/pir/cinn/sub_graphs/base.py
@@ -61,7 +61,10 @@ def train(self, net, to_static, with_prim=False, with_cinn=False):
                 )
             else:
                 net = paddle.jit.to_static(
-                    net(), full_graph=True, input_spec=self.input_specs
+                    net(),
+                    backend=None,
+                    full_graph=True,
+                    input_spec=self.input_specs,
                 )
         if self.with_train:
             net.train()
diff --git a/test/ir/pir/cinn/symbolic/test_decomp_inference_predictor_run.py b/test/ir/pir/cinn/symbolic/test_decomp_inference_predictor_run.py
@@ -58,6 +58,7 @@ def setUp(self):
                 ),
             ],
             full_graph=True,
+            backend=None,
         )
         paddle.jit.save(
             model,
diff --git a/test/ir/pir/cinn/symbolic/test_reshape_dyshape2stshape.py b/test/ir/pir/cinn/symbolic/test_reshape_dyshape2stshape.py
@@ -54,7 +54,11 @@ def train(self, net, to_static, with_cinn=False):
                     full_graph=True,
                 )
             else:
-                net = paddle.jit.to_static(net, full_graph=True)
+                net = paddle.jit.to_static(
+                    net,
+                    backend=None,
+                    full_graph=True,
+                )
         paddle.seed(123)
         outs = net(*self.inputs)
         return outs
diff --git a/test/ir/pir/cinn/symbolic/test_sub_graph_flatten.py b/test/ir/pir/cinn/symbolic/test_sub_graph_flatten.py
@@ -51,7 +51,11 @@ def train(self, net, to_static, with_prim=False, with_cinn=False):
                     full_graph=True,
                 )
             else:
-                net = paddle.jit.to_static(net, full_graph=True)
+                net = paddle.jit.to_static(
+                    net,
+                    backend=None,
+                    full_graph=True,
+                )
         paddle.seed(123)
         outs = net(*self.inputs)
         return outs
diff --git a/test/legacy_test/test_stack_op.py b/test/legacy_test/test_stack_op.py
@@ -396,7 +396,11 @@ def setUp(self):
     def test_list_single_tensor(self):
         expect = paddle.stack(self.x)
         paddle.base.core._set_prim_all_enabled(True)
-        st_model = paddle.jit.to_static(paddle.stack, full_graph=True)
+        st_model = paddle.jit.to_static(
+            paddle.stack,
+            backend=None,
+            full_graph=True,
+        )
         actual = st_model(self.x)
         np.testing.assert_allclose(expect, actual)
         paddle.enable_static()

Original file line number	Diff line number	Diff line change
`@@ -61,7 +61,10 @@ def train(self, net, to_static, with_prim=False, with_cinn=False):`
`61`	`61`	`)`
`62`	`62`	`else:`
`63`	`63`	`net = paddle.jit.to_static(`
`64`		`- net(), full_graph=True, input_spec=self.input_specs`
	`64`	`+ net(),`
	`65`	`+ backend=None,`
	`66`	`+ full_graph=True,`
	`67`	`+ input_spec=self.input_specs,`
`65`	`68`	`)`
`66`	`69`	`if self.with_train:`
`67`	`70`	`net.train()`
Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,7 @@ def setUp(self):`
`58`	`58`	`),`
`59`	`59`	`],`
`60`	`60`	`full_graph=True,`
	`61`	`+ backend=None,`
`61`	`62`	`)`
`62`	`63`	`paddle.jit.save(`
`63`	`64`	`model,`