|
|
@ -64,7 +64,7 @@ function(py_test_modules TARGET_NAME)
|
|
|
|
if (py_test_modules_SERIAL)
|
|
|
|
if (py_test_modules_SERIAL)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
|
|
|
|
set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
|
|
|
|
set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 350)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
endfunction()
|
|
|
|
endfunction()
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
@ -144,9 +144,7 @@ if(WITH_DISTRIBUTE)
|
|
|
|
set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
|
|
|
|
set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
|
|
|
|
set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
|
|
|
|
set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
|
|
|
|
py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
|
|
|
|
py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
|
|
|
|
set_tests_properties(test_dist_se_resnext PROPERTIES TIMEOUT 1000)
|
|
|
|
|
|
|
|
py_test_modules(test_dist_se_resnext_nccl MODULES test_dist_se_resnext_nccl SERIAL)
|
|
|
|
py_test_modules(test_dist_se_resnext_nccl MODULES test_dist_se_resnext_nccl SERIAL)
|
|
|
|
set_tests_properties(test_dist_se_resnext_nccl PROPERTIES TIMEOUT 1000)
|
|
|
|
|
|
|
|
# FIXME(typhoonzero): add these tests back
|
|
|
|
# FIXME(typhoonzero): add these tests back
|
|
|
|
# py_test_modules(test_dist_transformer MODULES test_dist_transformer)
|
|
|
|
# py_test_modules(test_dist_transformer MODULES test_dist_transformer)
|
|
|
|
# set_tests_properties(test_dist_transformer PROPERTIES TIMEOUT 1000)
|
|
|
|
# set_tests_properties(test_dist_transformer PROPERTIES TIMEOUT 1000)
|
|
|
|