[Dy2St] Enable test_lstm in PIR mode (#60343)

PaddlePaddle · Dec 26, 2023 · a726569 · a726569
1 parent 23808ae
commit a726569
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/dygraph_to_static/test_lstm.py b/test/dygraph_to_static/test_lstm.py
@@ -60,14 +60,14 @@ def tearDown(self):
 
     def run_lstm(self, to_static):
         with enable_to_static_guard(to_static):
-            paddle.static.default_main_program().random_seed = 1001
-            paddle.static.default_startup_program().random_seed = 1001
+            paddle.seed(1001)
 
             net = paddle.jit.to_static(Net(12, 2))
             x = paddle.zeros((2, 10, 12))
             y = net(x)
             return y.numpy()
 
+    @test_legacy_and_pt_and_pir
     def test_lstm_to_static(self):
         dygraph_out = self.run_lstm(to_static=False)
         static_out = self.run_lstm(to_static=True)