[Cleanup][C-8] Replace Program.random_seed (#61518)

PaddlePaddle · Feb 4, 2024 · 22cf91f · 22cf91f
1 parent b5f6235
commit 22cf91f
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 13 deletions.
diff --git a/test/legacy_test/test_fused_ec_moe_op.py b/test/legacy_test/test_fused_ec_moe_op.py
@@ -19,11 +19,10 @@
 
 import paddle
 import paddle.nn.functional as F
-from paddle.base.framework import default_main_program
 from paddle.incubate.nn.functional import fused_ec_moe
 from paddle.nn.layer.common import Linear
 
-default_main_program().random_seed = 42
+paddle.seed(42)
 
 
 class TestFusedEcMoEOp(OpTest):

diff --git a/test/legacy_test/test_fused_feedforward_op.py b/test/legacy_test/test_fused_feedforward_op.py
@@ -19,7 +19,6 @@
 import paddle
 import paddle.incubate.nn.functional as incubate_f
 import paddle.nn.functional as F
-from paddle.base.framework import default_main_program
 from paddle.nn.layer import transformer
 from paddle.nn.layer.common import Dropout, Linear
 from paddle.nn.layer.norm import LayerNorm
@@ -159,7 +158,7 @@ def FusedFFN(self):
         return out, x.grad
 
     def test_out_and_grad(self):
-        default_main_program().random_seed = 42
+        paddle.seed(42)
         base_out, base_grad = self.Base()
         fused_out, fused_grad = self.FusedFFN()
         np.testing.assert_allclose(
@@ -233,7 +232,7 @@ def run_fused_feedforward(
     ):
         main = paddle.static.Program()
         startup = paddle.static.Program()
-        main.random_seed = 42
+        paddle.seed(42)
         with paddle.static.program_guard(main, startup):
             x = paddle.static.data(
                 name='x',
@@ -314,7 +313,7 @@ def run_base_ffn(
     ):
         main = paddle.static.Program()
         startup = paddle.static.Program()
-        main.random_seed = 42
+        paddle.seed(42)
         with paddle.static.program_guard(main, startup):
             x = paddle.static.data(
                 name='x',

diff --git a/test/legacy_test/test_fused_feedforward_pass.py b/test/legacy_test/test_fused_feedforward_pass.py
@@ -103,9 +103,8 @@ def get_value(self, use_pass=False):
         )
 
         main_prog = paddle.static.Program()
-        main_prog.random_seed = 1234
         startup_prog = paddle.static.Program()
-        startup_prog.random_seed = 1234
+        paddle.seed(1234)
 
         with paddle.static.program_guard(main_prog, startup_prog):
             data = paddle.static.data(

diff --git a/test/legacy_test/test_fused_multi_transformer_int8_op.py b/test/legacy_test/test_fused_multi_transformer_int8_op.py
@@ -21,12 +21,11 @@
 import paddle.nn.functional as F
 from paddle import _legacy_C_ops, tensor
 from paddle.base import core
-from paddle.base.framework import default_main_program
 from paddle.nn.layer.common import Dropout
 from paddle.nn.layer.norm import LayerNorm
 from paddle.nn.layer.transformer import _convert_attention_mask
 
-default_main_program().random_seed = 42
+paddle.seed(42)
 np.random.seed(0)
 
 

diff --git a/test/xpu/test_fused_feedforward_op_xpu.py b/test/xpu/test_fused_feedforward_op_xpu.py
@@ -20,7 +20,6 @@
 import paddle
 import paddle.incubate.nn.functional as incubate_f
 import paddle.nn.functional as F
-from paddle.base.framework import default_main_program
 from paddle.nn.layer import transformer
 from paddle.nn.layer.common import Dropout, Linear
 from paddle.nn.layer.norm import LayerNorm
@@ -165,7 +164,7 @@ def FusedFFN(self):
             return out, x.grad
 
         def test_out_and_grad(self):
-            default_main_program().random_seed = 42
+            paddle.seed(42)
             base_out, base_grad = self.Base()
             fused_out, fused_grad = self.FusedFFN()
             np.testing.assert_allclose(
@@ -199,7 +198,7 @@ def getShape(self):
 class APITestStaticFusedFFN(unittest.TestCase):
     def test_static(self):
         paddle.enable_static()
-        default_main_program().random_seed = 42
+        paddle.seed(42)
         dtype = "float32"
         layer_norm_dtype = "float32"
         batch_size = 1