Commit 114d156d authored by Gregor Fabritius's avatar Gregor Fabritius Committed by Davis E. King

Fixed cudnn_hint_path detection for OSX where CUDA_CUBLAS_LIBRARIES is a list,…

Fixed cudnn_hint_path detection for OSX where CUDA_CUBLAS_LIBRARIES is a list, not a single string. (#230)
parent 29467871
...@@ -7,7 +7,7 @@ find_path(cudnn_include cudnn.h ...@@ -7,7 +7,7 @@ find_path(cudnn_include cudnn.h
PATHS /usr/local ENV CPATH PATHS /usr/local ENV CPATH
PATH_SUFFIXES include PATH_SUFFIXES include
) )
get_filename_component(cudnn_hint_path ${CUDA_CUBLAS_LIBRARIES} PATH) get_filename_component(cudnn_hint_path "${CUDA_CUBLAS_LIBRARIES}" PATH)
find_library(cudnn cudnn find_library(cudnn cudnn
HINTS ${cudnn_hint_path} ENV CUDNN_LIBRARY_DIR ENV CUDNN_HOME HINTS ${cudnn_hint_path} ENV CUDNN_LIBRARY_DIR ENV CUDNN_HOME
PATHS /usr/local /usr/local/cuda ENV LD_LIBRARY_PATH PATHS /usr/local /usr/local/cuda ENV LD_LIBRARY_PATH
......
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