|
|
|
@ -229,6 +229,7 @@ op_library(recurrent_op DEPS executor)
|
|
|
|
|
op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
|
|
|
|
|
op_library(cos_sim_op DEPS cos_sim_functor)
|
|
|
|
|
op_library(parallel_do_op DEPS executor)
|
|
|
|
|
op_library(ckpt_save_op DEPS lod_tensor)
|
|
|
|
|
|
|
|
|
|
if (WITH_GPU)
|
|
|
|
|
op_library(conv_op DEPS vol2col depthwise_conv im2col)
|
|
|
|
@ -277,5 +278,6 @@ cc_test(beam_search_op_test SRCS beam_search_op_test.cc DEPS lod_tensor beam_sea
|
|
|
|
|
cc_test(strided_memcpy_test SRCS strided_memcpy_test.cc DEPS tensor memory)
|
|
|
|
|
cc_test(save_load_op_test SRCS save_load_op_test.cc DEPS save_op load_op)
|
|
|
|
|
cc_test(save_load_combine_op_test SRCS save_load_combine_op_test.cc DEPS save_combine_op load_combine_op)
|
|
|
|
|
cc_test(ckpt_save_op_test SRCS ckpt_save_op_test.cc DEPS ckpt_save_op)
|
|
|
|
|
nv_test(nccl_op_test SRCS nccl_op_test.cu.cc DEPS nccl_op gpu_info device_context)
|
|
|
|
|
nv_test(dropout_op_test SRCS dropout_op_test.cc DEPS dropout_op tensor)
|
|
|
|
|