|
|
|
@ -16,12 +16,10 @@ add_simple_unittest(test_CpuGpuVector)
|
|
|
|
|
add_simple_unittest(test_Allocator)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
if(COMPILER_SUPPORT_CXX11)
|
|
|
|
|
CUDA_ADD_EXECUTABLE(test_Tensor test_Tensor.cu)
|
|
|
|
|
link_paddle_test(test_Tensor)
|
|
|
|
|
CUDA_ADD_EXECUTABLE(test_lazyAssign test_lazyAssign.cu)
|
|
|
|
|
link_paddle_test(test_lazyAssign)
|
|
|
|
|
endif()
|
|
|
|
|
CUDA_ADD_EXECUTABLE(test_Tensor test_Tensor.cu)
|
|
|
|
|
link_paddle_test(test_Tensor)
|
|
|
|
|
CUDA_ADD_EXECUTABLE(test_lazyAssign test_lazyAssign.cu)
|
|
|
|
|
link_paddle_test(test_lazyAssign)
|
|
|
|
|
else()
|
|
|
|
|
compile_cu_as_cpp(test_Tensor.cu)
|
|
|
|
|
add_unittest(test_Tensor test_Tensor.cu)
|
|
|
|
|