Skip to content

GitLab

Commit f442ef4e authored by Christopher Jan-Steffen Brix's avatar Christopher Jan-Steffen Brix
Browse files

Updated test

parent 870ee340
Pipeline #110327 passed with stages
in 4 minutes and 2 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/test/resources/target_code/CMakeLists.txt
View file @ f442ef4e
......@@ -17,11 +17,11 @@ set(LIBS ${LIBS} ${Caffe2_LIBRARIES})
# additional commands
set(LIBS ${LIBS} -lprotobuf -lglog -lgflags)
find_package(CUDA)
set(INCLUDE_DIRS ${INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS})
set(LIBS ${LIBS} ${CUDA_LIBRARIES} ${CUDA_curand_LIBRARY})
if(CUDA_FOUND)
set(LIBS ${LIBS} caffe2 caffe2_gpu)
set(INCLUDE_DIRS ${INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS})
set(LIBS ${LIBS} ${CUDA_LIBRARIES} ${CUDA_curand_LIBRARY})
else()
set(LIBS ${LIBS} caffe2)
endif()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment