-
Notifications
You must be signed in to change notification settings - Fork 134
Open
Description
My machine has cuda install at: /usr/local/cuda; it's fine when I build with cuda. However, if I don't want to build it with cuda
cmake .. -DNO_CUDA=1
I get following errors:
[ 74%] Building CXX object src/ATen/test/CMakeFiles/cudnn_test.dir/cudnn_test.cpp.o
In file included from /home/rook/ATen2/src/ATen/cudnn/cudnn-wrapper.h:3:0,
from /home/rook/ATen2/src/ATen/cudnn/Exceptions.h:3,
from /home/rook/ATen2/src/ATen/cudnn/Descriptors.h:3,
from /home/rook/ATen2/src/ATen/test/cudnn_test.cpp:2:
/usr/include/cudnn.h:63:26: fatal error: driver_types.h: No such file or directory
compilation terminated.
Because of this
if(NOT NO_CUDA)
cuda_add_executable(integer_divider_test integer_divider_test.cu)
target_link_libraries(integer_divider_test ATen)
endif()
if (CUDNN_FOUND)
add_executable(cudnn_test cudnn_test.cpp)
target_link_libraries(cudnn_test ATen)
endif()
in src/ATen/test/CMakeLists.txt
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels