|
|
|
@ -43,12 +43,10 @@ list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dist_train)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
|
|
|
|
|
# TODO(wuyi): this test hungs on CI, will add it back later
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_listen_and_serv_op)
|
|
|
|
|
foreach(TEST_OP ${TEST_OPS})
|
|
|
|
|
py_test_modules(${TEST_OP} MODULES ${TEST_OP})
|
|
|
|
|
endforeach(TEST_OP)
|
|
|
|
|
py_test_modules(test_warpctc_op MODULES test_warpctc_op ENVS FLAGS_warpctc_dir=${WARPCTC_LIB_DIR} SERIAL)
|
|
|
|
|
py_test_modules(test_dist_train MODULES test_dist_train SERIAL)
|
|
|
|
|
# FIXME(Yancey1989): this test would cost much more time on CUDAPlace
|
|
|
|
|
# since load cudnn libraries, so we use a longer timeout to make this
|
|
|
|
|
# unit test stability.
|
|
|
|
|
set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 30)
|
|
|
|
|