fix error when setting ut timeout value (#28696)

5 years ago · e880c90c5a
parent 628fb29c1b
commit e880c90c5a
1 changed files with 6 additions and 3 deletions
--- a/python/paddle/fluid/tests/unittests/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/CMakeLists.txt
@ -621,11 +621,14 @@ endif()
 if (WITH_DISTRIBUTE)
    set_tests_properties(test_communicator_half_async PROPERTIES TIMEOUT 120)
 endif()
+
 if (WITH_DISTRIBUTE AND NOT APPLE)
+    if(WITH_GPU)
+        set_tests_properties(test_c_comm_init_op PROPERTIES TIMEOUT 120)
+        set_tests_properties(test_fleet_checkpoint PROPERTIES TIMEOUT 120)
+        set_tests_properties(test_launch PROPERTIES TIMEOUT 120)
+    endif()
    set_tests_properties(test_fleet_launch PROPERTIES TIMEOUT 120)
-    set_tests_properties(test_c_comm_init_op PROPERTIES TIMEOUT 120)
-    set_tests_properties(test_fleet_checkpoint PROPERTIES TIMEOUT 120)
-    set_tests_properties(test_launch PROPERTIES TIMEOUT 120)
 endif()

 # setting timeout value as 15S