|
|
|
@ -5,11 +5,11 @@ proto_library(framework_proto SRCS framework.proto)
|
|
|
|
|
cc_library(ddim SRCS ddim.cc DEPS eigen3 boost)
|
|
|
|
|
cc_test(ddim_test SRCS ddim_test.cc DEPS ddim)
|
|
|
|
|
nv_test(dim_test SRCS dim_test.cu DEPS ddim)
|
|
|
|
|
cc_library(data_type SRCS data_type.cc DEPS framework_proto ddim)
|
|
|
|
|
cc_library(data_type SRCS data_type.cc DEPS framework_proto ddim device_context)
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
nv_library(tensor SRCS tensor.cc tensor_util.cu DEPS place memory device_context data_type)
|
|
|
|
|
nv_library(tensor SRCS tensor.cc tensor_util.cu DEPS place memory data_type)
|
|
|
|
|
else()
|
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS place memory device_context data_type)
|
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS place memory data_type)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
cc_test(tensor_test SRCS tensor_test.cc DEPS tensor)
|
|
|
|
|