|
|
|
@ -41,8 +41,8 @@ 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_parallel_executor_crf)
|
|
|
|
|
#list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
|
|
|
|
|
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})
|
|
|
|
@ -50,3 +50,5 @@ foreach(TEST_OP ${TEST_OPS})
|
|
|
|
|
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)
|
|
|
|
|
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)
|
|
|
|
|