|
|
|
@ -7,9 +7,9 @@ cc_test(ddim_test SRCS ddim_test.cc DEPS ddim)
|
|
|
|
|
nv_test(dim_test SRCS dim_test.cu DEPS ddim)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
nv_library(tensor SRCS tensor.cc tensor_util.cu DEPS ddim place paddle_memory device_context framework_proto)
|
|
|
|
|
nv_library(tensor SRCS tensor.cc tensor_util.cu DEPS ddim place memory device_context framework_proto)
|
|
|
|
|
else()
|
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS ddim place paddle_memory device_context framework_proto)
|
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS ddim place memory device_context framework_proto)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
cc_test(tensor_test SRCS tensor_test.cc DEPS tensor)
|
|
|
|
@ -21,9 +21,9 @@ endif()
|
|
|
|
|
|
|
|
|
|
cc_test(eigen_test SRCS eigen_test.cc DEPS tensor)
|
|
|
|
|
|
|
|
|
|
nv_test(mixed_vector_test SRCS mixed_vector_test.cu DEPS place paddle_memory device_context init)
|
|
|
|
|
nv_test(mixed_vector_test SRCS mixed_vector_test.cu DEPS place memory device_context init)
|
|
|
|
|
cc_library(lod_tensor SRCS lod_tensor.cc DEPS ddim place tensor framework_proto recordio)
|
|
|
|
|
cc_test(lod_tensor_test SRCS lod_tensor_test.cc DEPS lod_tensor paddle_memory)
|
|
|
|
|
cc_test(lod_tensor_test SRCS lod_tensor_test.cc DEPS lod_tensor memory)
|
|
|
|
|
nv_test(lod_tensor_gpu_test SRCS lod_tensor_test.cu DEPS lod_tensor init)
|
|
|
|
|
|
|
|
|
|
cc_library(reader SRCS reader.cc DEPS lod_tensor ddim)
|
|
|
|
|