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

Wanglongzhi2001 · Jan 7, 2024 · b66472e · b66472e
1 parent aec1a81
commit b66472e
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)