|
|
|
@ -28,6 +28,8 @@ function(py_test_modules TARGET_NAME)
|
|
|
|
|
endif()
|
|
|
|
|
endfunction()
|
|
|
|
|
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_sequence_expand)
|
|
|
|
|
|
|
|
|
|
# test time consuming OPs in a separate process for expliot parallism
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dyn_rnn)
|
|
|
|
@ -63,6 +65,8 @@ else()
|
|
|
|
|
endforeach(TEST_OP)
|
|
|
|
|
endif(WITH_FAST_BUNDLE_TEST)
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
|
py_test_modules(test_sequence_expand MODULES test_sequence_expand)
|
|
|
|
|
# tests with high overhead
|
|
|
|
|
py_test_modules(test_warpctc_op MODULES test_warpctc_op ENVS FLAGS_warpctc_dir=${WARPCTC_LIB_DIR})
|
|
|
|
|
py_test_modules(test_train_dyn_rnn MODULES test_dyn_rnn)
|
|
|
|
|