|
|
|
@ -325,6 +325,7 @@ function(parallel_bash_test_modules TARGET_NAME)
|
|
|
|
|
endfunction()
|
|
|
|
|
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_feed_data_check_shape_type)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_fetch_lod_tensor_array)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
|
|
|
|
|
list(REMOVE_ITEM TEST_OPS test_parallel_executor_profiler)
|
|
|
|
@ -538,10 +539,12 @@ if(WIN32)
|
|
|
|
|
py_test_modules(test_parallel_executor_transformer_auto_growth MODULES test_parallel_executor_transformer_auto_growth ENVS FLAGS_allocator_strategy=auto_growth CUDA_VISIBLE_DEVICES=0)
|
|
|
|
|
py_test_modules(test_fuse_all_reduce_pass MODULES test_fuse_all_reduce_pass ENVS CUDA_VISIBLE_DEVICES=0)
|
|
|
|
|
py_test_modules(test_feed_data_check_shape_type MODULES test_feed_data_check_shape_type ENVS CUDA_VISIBLE_DEVICES=0)
|
|
|
|
|
py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array ENVS CUDA_VISIBLE_DEVICES=0)
|
|
|
|
|
else()
|
|
|
|
|
py_test_modules(test_parallel_executor_transformer_auto_growth MODULES test_parallel_executor_transformer_auto_growth ENVS FLAGS_allocator_strategy=auto_growth)
|
|
|
|
|
py_test_modules(test_fuse_all_reduce_pass MODULES test_fuse_all_reduce_pass)
|
|
|
|
|
py_test_modules(test_feed_data_check_shape_type MODULES test_feed_data_check_shape_type)
|
|
|
|
|
py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
py_test_modules(test_data_norm_op MODULES test_data_norm_op)
|
|
|
|
|