|
|
|
@ -43,11 +43,11 @@ cc_library(backward SRCS backward.cc DEPS net_op)
|
|
|
|
|
cc_test(backward_test SRCS backward_test.cc DEPS backward recurrent_op device_context)
|
|
|
|
|
|
|
|
|
|
cc_library(executor SRCS executor.cc DEPS op_registry device_context scope framework_proto backward ${GLOB_OP_LIB})
|
|
|
|
|
#if(WITH_GPU)
|
|
|
|
|
# nv_test(executor_test SRCS executor_test.cc DEPS executor)
|
|
|
|
|
#else()
|
|
|
|
|
# cc_test(executor_test SRCS executor_test.cc DEPS executor)
|
|
|
|
|
#endif()
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
nv_test(executor_test SRCS executor_test.cc DEPS executor)
|
|
|
|
|
else()
|
|
|
|
|
cc_test(executor_test SRCS executor_test.cc DEPS executor)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
cc_library(tensor_array SRCS tensor_array.cc DEPS lod_tensor)
|
|
|
|
|
cc_test(tensor_array_test SRCS tensor_array_test.cc DEPS tensor_array place)
|
|
|
|
|