[PIR] fix some pir test case (#65998)

wanghuancoder · web-flow · commit 449b6ac7907f · 2024-07-15T10:37:23.000+08:00
* fix some pir test case

* refine
diff --git a/paddle/fluid/prim/api/auto_code_generated/eager_gen.py b/paddle/fluid/prim/api/auto_code_generated/eager_gen.py
@@ -158,7 +158,7 @@ def parse_input_and_attr(self, inputs_list, attrs_list):
             'Scalar(int)': 'const Scalar&',
             'Scalar(int64_t)': 'const Scalar&',
             'Scalar(float)': 'const Scalar&',
-            'Scalar(dobule)': 'const Scalar&',
+            'Scalar(double)': 'const Scalar&',
             'Scalar[]': 'const std::vector<phi::Scalar>&',
             'int': 'int',
             'int32_t': 'int32_t',
diff --git a/test/deprecated/legacy_test/test_switch_autotune.py b/test/deprecated/legacy_test/test_switch_autotune.py
@@ -121,49 +121,50 @@ def test_disable_autotune(self):
 
 class TestStaticAutoTuneStatus(TestAutoTune):
     def run_program(self, enable_autotune):
-        paddle.enable_static()
-
-        data_shape = [1, 1, 8, 8]
-        main_program = paddle.static.Program()
-        startup_program = paddle.static.Program()
-        with paddle.static.program_guard(main_program, startup_program):
-            data = paddle.static.data(
-                name='X', shape=data_shape, dtype='float32'
+        with paddle.pir_utils.OldIrGuard():
+            paddle.enable_static()
+
+            data_shape = [1, 1, 8, 8]
+            main_program = paddle.static.Program()
+            startup_program = paddle.static.Program()
+            with paddle.static.program_guard(main_program, startup_program):
+                data = paddle.static.data(
+                    name='X', shape=data_shape, dtype='float32'
+                )
+                net = SimpleNet()
+                loss = static_program(net, data)
+            place = (
+                paddle.CUDAPlace(0)
+                if paddle.base.core.is_compiled_with_cuda()
+                else paddle.CPUPlace()
             )
-            net = SimpleNet()
-            loss = static_program(net, data)
-        place = (
-            paddle.CUDAPlace(0)
-            if paddle.base.core.is_compiled_with_cuda()
-            else paddle.CPUPlace()
-        )
-        exe = paddle.static.Executor(place)
-        exe.run(startup_program)
-        x = np.random.random(size=data_shape).astype('float32')
-
-        # Node(tizheng): warmup run to make sure the following runs
-        # are in the same thread. Necessary for CUDNNv8 tests
-        exe.run(program=main_program, feed={'X': x}, fetch_list=[loss])
+            exe = paddle.static.Executor(place)
+            exe.run(startup_program)
+            x = np.random.random(size=data_shape).astype('float32')
 
-        self.set_flags(enable_autotune)
-        if enable_autotune:
-            config = {"kernel": {"enable": True, "tuning_range": [1, 2]}}
-            tfile = tempfile.NamedTemporaryFile(mode="w+", delete=False)
-            json.dump(config, tfile)
-            tfile.close()
-            paddle.incubate.autotune.set_config(tfile.name)
-            os.remove(tfile.name)
-        else:
-            paddle.incubate.autotune.set_config(
-                config={"kernel": {"enable": False, "tuning_range": [1, 2]}}
-            )
-
-        for i in range(3):
+            # Node(tizheng): warmup run to make sure the following runs
+            # are in the same thread. Necessary for CUDNNv8 tests
             exe.run(program=main_program, feed={'X': x}, fetch_list=[loss])
-            status = paddle.base.core.autotune_status()
-            expected_res = self.get_expected_res(i, enable_autotune)
-            self.check_status(expected_res)
-        paddle.disable_static()
+
+            self.set_flags(enable_autotune)
+            if enable_autotune:
+                config = {"kernel": {"enable": True, "tuning_range": [1, 2]}}
+                tfile = tempfile.NamedTemporaryFile(mode="w+", delete=False)
+                json.dump(config, tfile)
+                tfile.close()
+                paddle.incubate.autotune.set_config(tfile.name)
+                os.remove(tfile.name)
+            else:
+                paddle.incubate.autotune.set_config(
+                    config={"kernel": {"enable": False, "tuning_range": [1, 2]}}
+                )
+
+            for i in range(3):
+                exe.run(program=main_program, feed={'X': x}, fetch_list=[loss])
+                status = paddle.base.core.autotune_status()
+                expected_res = self.get_expected_res(i, enable_autotune)
+                self.check_status(expected_res)
+            paddle.disable_static()
 
     def func_enable_autotune(self):
         self.run_program(enable_autotune=True)
diff --git a/test/deprecated/legacy_test/test_transformer_api.py b/test/deprecated/legacy_test/test_transformer_api.py
@@ -283,8 +283,15 @@ class TestTransformer(unittest.TestCase):
     def test_multi_head_attention(self):
         def multihead_attention_test_helper(self_attention, cache):
             paddle.seed(2020)
-            paddle.framework.random._manual_program_seed(2020)
-            # self_attention|cross_attention, cache|No cache
+            if paddle.framework.use_pir_api():
+                with paddle.pir_utils.OldIrGuard():
+                    # Note: dygraph use self.main_program.global_block().create_parameter(), it's need manual seed to old Program
+                    paddle.framework.random._manual_program_seed(2020)
+                paddle.framework.random._manual_program_seed(2020)
+            else:
+                paddle.framework.random._manual_program_seed(
+                    2020
+                )  # self_attention|cross_attention, cache|No cache
             with base.dygraph.guard(base.CPUPlace()):
                 # generate params for multi_head_attention
                 (
@@ -401,7 +408,15 @@ def multihead_attention_test_helper(self_attention, cache):
     def test_transformer_encoder_layer(self):
         with base.dygraph.guard(base.CPUPlace()):
             paddle.framework.seed(2020)
-            paddle.framework.random._manual_program_seed(2020)
+            if paddle.framework.use_pir_api():
+                with paddle.pir_utils.OldIrGuard():
+                    # Note: dygraph use self.main_program.global_block().create_parameter(), it's need manual seed to old Program
+                    paddle.framework.random._manual_program_seed(2020)
+                paddle.framework.random._manual_program_seed(2020)
+            else:
+                paddle.framework.random._manual_program_seed(
+                    2020
+                )  # self_attention|cross_attention, cache|No cache
 
             ffn_fc1_act = "relu"
             # 1.generate basic params
@@ -466,7 +481,15 @@ def test_transformer_encoder_layer(self):
     def test_transformer_encoder_layer_attr_1(self):
         with base.dygraph.guard(base.CPUPlace()):
             paddle.framework.seed(2020)
-            paddle.framework.random._manual_program_seed(2020)
+            if paddle.framework.use_pir_api():
+                with paddle.pir_utils.OldIrGuard():
+                    # Note: dygraph use self.main_program.global_block().create_parameter(), it's need manual seed to old Program
+                    paddle.framework.random._manual_program_seed(2020)
+                paddle.framework.random._manual_program_seed(2020)
+            else:
+                paddle.framework.random._manual_program_seed(
+                    2020
+                )  # self_attention|cross_attention, cache|No cache
 
             ffn_fc1_act = "relu"
             # 1.generate basic params
