fix the stack op bug in the old IR (PaddlePaddle#66457)

inaomIIsfarell · Jul 31, 2024 · dc23d84 · dc23d84
1 parent a934541
commit dc23d84
Showing 1 changed file with 14 additions and 13 deletions.
diff --git a/test/legacy_test/test_stack_op.py b/test/legacy_test/test_stack_op.py
@@ -390,19 +390,20 @@ def test_dygraph(self):
         paddle.enable_static()
 
 
-# class TestStackListOfSingleTensor(unittest.TestCase):
-#     def setUp(self):
-#         paddle.disable_static()
-#         paddle.seed(2022)
-#         self.x = [paddle.randn((4, 2, 6), dtype="float32")]
-
-#     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)
-#         actual = st_model(self.x)
-#         np.testing.assert_allclose(expect, actual)
-#         paddle.enable_static()
+class TestStackListOfSingleTensor(unittest.TestCase):
+    def setUp(self):
+        paddle.disable_static()
+        paddle.seed(2022)
+        self.x = [paddle.randn((4, 2, 6), dtype="float32")]
+        self.x[0].stop_gradient = False
+
+    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)
+        actual = st_model(self.x)
+        np.testing.assert_allclose(expect, actual)
+        paddle.enable_static()
 
 
 class TestPrimStackGrad(unittest.TestCase):