|
|
|
@ -37,66 +37,6 @@ 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)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_mul_op)
|
|
|
|
|
|
|
|
|
|
# tests that need to be run in separate process.
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_multihead_attention)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_calc_gradient)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_while_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_lod_array_length_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_reorder_lod_tensor)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_profiler)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_nvprof)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_normalization_wrapper)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_executor_and_mul)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_assign_value_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_array_read_write_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_lod_rank_table)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_weight_normalization)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_conditional_block)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parameter)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_registry)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_fetch_var)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dynrnn_static_input)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_dist_train)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_network_with_dtype)
|
|
|
|
|
|
|
|
|
|
foreach(TEST_OP ${TEST_OPS})
|
|
|
|
|
py_test_modules(${TEST_OP} MODULES ${TEST_OP})
|
|
|
|
|
endforeach(TEST_OP)
|
|
|
|
|
|
|
|
|
|
#
|
|
|
|
|
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)
|
|
|
|
|
py_test_modules(test_mul_op MODULES test_mul_op)
|
|
|
|
|
py_test_modules(test_network_with_dtype MODULES test_network_with_dtype)
|
|
|
|
|
|
|
|
|
|
# tests that need to be run in separate process.
|
|
|
|
|
py_test_modules(test_multihead_attention MODULES test_multihead_attention)
|
|
|
|
|
py_test_modules(test_calc_gradient MODULES test_calc_gradient)
|
|
|
|
|
py_test_modules(test_while_op MODULES test_while_op)
|
|
|
|
|
py_test_modules(test_lod_array_length_op MODULES test_lod_array_length_op)
|
|
|
|
|
py_test_modules(test_reorder_lod_tensor MODULES test_reorder_lod_tensor)
|
|
|
|
|
py_test_modules(test_profiler MODULES test_profiler)
|
|
|
|
|
py_test_modules(test_nvprof MODULES test_nvprof)
|
|
|
|
|
py_test_modules(test_normalization_wrapper MODULES test_normalization_wrapper)
|
|
|
|
|
py_test_modules(test_executor_and_mul MODULES test_executor_and_mul)
|
|
|
|
|
py_test_modules(test_assign_value_op MODULES test_assign_value_op)
|
|
|
|
|
py_test_modules(test_array_read_write_op MODULES test_array_read_write_op)
|
|
|
|
|
py_test_modules(test_lod_rank_table MODULES test_lod_rank_table)
|
|
|
|
|
py_test_modules(test_weight_normalization MODULES test_weight_normalization)
|
|
|
|
|
py_test_modules(test_conditional_block MODULES test_conditional_block)
|
|
|
|
|
py_test_modules(test_parameter MODULES test_parameter)
|
|
|
|
|
py_test_modules(test_registry MODULES test_registry)
|
|
|
|
|
py_test_modules(test_fetch_var MODULES test_fetch_var)
|
|
|
|
|
py_test_modules(test_dynrnn_static_input MODULES test_dynrnn_static_input)
|
|
|
|
|
py_test_modules(test_parallel_op MODULES test_parallel_op)
|
|
|
|
|
py_test_modules(test_dist_train MODULES test_dist_train)
|
|
|
|
|