|
|
@ -6,7 +6,7 @@ cc_test(ddim_test SRCS ddim_test.cc DEPS ddim)
|
|
|
|
nv_test(dim_test SRCS dim_test.cu DEPS ddim)
|
|
|
|
nv_test(dim_test SRCS dim_test.cu DEPS ddim)
|
|
|
|
|
|
|
|
|
|
|
|
if (WITH_GPU)
|
|
|
|
if (WITH_GPU)
|
|
|
|
nv_binary(tensor SRCS tensor.cc tensor_util.cu DEPS ddim place paddle_memory device_context)
|
|
|
|
nv_library(tensor SRCS tensor.cc tensor_util.cu DEPS ddim place paddle_memory device_context)
|
|
|
|
else()
|
|
|
|
else()
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS ddim place paddle_memory device_context)
|
|
|
|
cc_library(tensor SRCS tensor.cc tensor_util.cc DEPS ddim place paddle_memory device_context)
|
|
|
|
endif ()
|
|
|
|
endif ()
|
|
|
|