From e880c90c5a091e4a331b09b43987142b02f61ac1 Mon Sep 17 00:00:00 2001
From: YUNSHEN XIE <1084314248@qq.com>
Date: Wed, 18 Nov 2020 16:44:31 +0800
Subject: [PATCH] fix error when setting ut timeout value (#28696)

---
 python/paddle/fluid/tests/unittests/CMakeLists.txt | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/CMakeLists.txt b/python/paddle/fluid/tests/unittests/CMakeLists.txt
index 6e78f7d9014..e527ba613ba 100644
--- 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
-- 
GitLab