From d1bd3fdefc9ec5a2c8d3746ab833dabd9f841948 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 11 May 2018 19:10:24 +0800 Subject: [PATCH] add build and test make --- paddle/fluid/operators/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/operators/CMakeLists.txt b/paddle/fluid/operators/CMakeLists.txt index 256aded8ca2..a6c7690d6b7 100644 --- a/paddle/fluid/operators/CMakeLists.txt +++ b/paddle/fluid/operators/CMakeLists.txt @@ -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) -- GitLab