|
|
|
@ -115,7 +115,7 @@ if(WITH_GPU)
|
|
|
|
|
set(ENV{CUDNN_INCLUDE_DIR} ${CUDNN_INCLUDE_DIR})
|
|
|
|
|
get_filename_component(CUDNN_LIBRARY_DIR ${CUDNN_LIBRARY} DIRECTORY)
|
|
|
|
|
set(ENV{CUDNN_LIBRARY} ${CUDNN_LIBRARY_DIR})
|
|
|
|
|
message(STATUS "Set Anakin CUDNN_LIBRARY_DIR:" ${CUDNN_LIBRARY_DIR})
|
|
|
|
|
message(STATUS "Set Anakin build arg, CUDNN_LIBRARY_DIR: " ${CUDNN_LIBRARY_DIR})
|
|
|
|
|
endif()
|
|
|
|
|
elseif(WITH_AMD_GPU)
|
|
|
|
|
add_definitions(-DPADDLE_WITH_HIP)
|
|
|
|
|