|
|
|
@ -46,6 +46,7 @@ function(py_test_modules TARGET_NAME)
|
|
|
|
|
endfunction()
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dist_train)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dist_transpiler)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext)
|
|
|
|
@ -61,11 +62,12 @@ if(WITH_DISTRIBUTE)
|
|
|
|
|
set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 20)
|
|
|
|
|
set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
|
|
|
|
|
set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
|
|
|
|
|
py_test_modules(test_dist_transpiler MODULES test_dist_transpiler)
|
|
|
|
|
py_test_modules(test_dist_transformer MODULES test_dist_transformer SERIAL)
|
|
|
|
|
py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext SERIAL)
|
|
|
|
|
endif()
|
|
|
|
|
py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf SERIAL)
|
|
|
|
|
py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed SERIAL)
|
|
|
|
|
set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 150)
|
|
|
|
|
py_test_modules(test_dist_transformer MODULES test_dist_transformer SERIAL)
|
|
|
|
|
py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext SERIAL)
|
|
|
|
|
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL)
|
|
|
|
|
py_test_modules(test_image_classification_resnet MODULES test_image_classification_resnet SERIAL)
|
|
|
|
|