diff --git a/test/deprecated/legacy_test/test_slice_var.py b/test/legacy_test/test_slice_var.py similarity index 85% rename from test/deprecated/legacy_test/test_slice_var.py rename to test/legacy_test/test_slice_var.py index 1ed9d4dfa9e273..6e73d5bdc566e7 100644 --- a/test/deprecated/legacy_test/test_slice_var.py +++ b/test/legacy_test/test_slice_var.py @@ -15,6 +15,7 @@ import random import unittest +import paddle from paddle import base from paddle.distributed.transpiler.distribute_transpiler import slice_variable @@ -22,14 +23,15 @@ class TestSliceVar(unittest.TestCase): def check_slice_output(self, shapes, expected_sizes, min_size): var_list = [] - program = base.Program() - for shape in shapes: - var = program.global_block().create_var( - name=str(random.randint(10000, 99999)), - persistable=True, - shape=shape, - ) - var_list.append(var) + with paddle.pir_utils.OldIrGuard(): + program = base.Program() + for shape in shapes: + var = program.global_block().create_var( + name=str(random.randint(10000, 99999)), + persistable=True, + shape=shape, + ) + var_list.append(var) blocks = slice_variable(var_list, 10, min_size) all_sizes = [] for s in expected_sizes: