【Fix PIR Unittest No.14 BUAA】Fix some test case in PIR (PaddlePaddle#…

…66152) * Fix some bugs in test_logcumsumexp_op when PIR not support .name. * Add TIMEOUT SETTING for test_logsumcumexp_op.
inaomIIsfarell · Jul 31, 2024 · 206ba2f · 206ba2f
1 parent 22a6aa3
commit 206ba2f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/test/deprecated/legacy_test/CMakeLists.txt b/test/deprecated/legacy_test/CMakeLists.txt
@@ -467,7 +467,6 @@ endif()
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP})
 endforeach()
-set_tests_properties(test_logcumsumexp_op PROPERTIES TIMEOUT 30)
 
 py_test_modules(
   test_imperative_ocr_attention_model MODULES

diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt
@@ -543,6 +543,7 @@ endif()
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP})
 endforeach()
+set_tests_properties(test_logcumsumexp_op PROPERTIES TIMEOUT 30)
 if(WITH_GPU
    OR WITH_XPU
    OR WITH_ASCEND

diff --git a/...cated/legacy_test/test_logcumsumexp_op.py → test/legacy_test/test_logcumsumexp_op.py b/...cated/legacy_test/test_logcumsumexp_op.py → test/legacy_test/test_logcumsumexp_op.py
@@ -201,10 +201,11 @@ def test_gpu(self):
 
     def test_name(self):
         paddle.enable_static()
-        with base.program_guard(base.Program()):
-            x = paddle.static.data('x', [3, 4])
-            y = paddle.logcumsumexp(x, name='out')
-            self.assertTrue('out' in y.name)
+        with paddle.pir_utils.OldIrGuard():
+            with base.program_guard(base.Program()):
+                x = paddle.static.data('x', [3, 4])
+                y = paddle.logcumsumexp(x, name='out')
+                self.assertTrue('out' in y.name)
         paddle.disable_static()
 
     @test_with_pir_api