[Dy2St] Increase test_resnet_amp timeout (#62835)

PaddlePaddle · Mar 19, 2024 · f998342 · f998342
1 parent 23c9830
commit f998342
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/dygraph_to_static/CMakeLists.txt b/test/dygraph_to_static/CMakeLists.txt
@@ -16,8 +16,8 @@ if(WITH_PYTHON)
 endif()
 
 if(WIN32 AND NOT WITH_GPU)
-  list(REMOVE_ITEM TEST_OPS test_resnet_amp
-  )# disable on Windows CPU CI for timeout
+  # disable on Windows CPU CI for timeout
+  list(REMOVE_ITEM TEST_OPS test_resnet_amp)
 endif()
 
 if(NOT WITH_GPU)
@@ -48,6 +48,10 @@ set_tests_properties(test_bmn PROPERTIES TIMEOUT 300)
 set_tests_properties(test_loop PROPERTIES TIMEOUT 180)
 set_tests_properties(test_mnist_amp PROPERTIES TIMEOUT 240)
 
+if(TEST test_resnet_amp)
+  set_tests_properties(test_resnet_amp PROPERTIES TIMEOUT 240)
+endif()
+
 if(NOT WIN32)
   set_tests_properties(test_tsm PROPERTIES TIMEOUT 900)
 endif()