Unverified Commit 6b8fae68 authored by Jai Menon's avatar Jai Menon Committed by GitHub

Merge branch 'master' into jmenon/warnings

parents 535125f8 05d80db3
......@@ -113,7 +113,7 @@ endif()
#-----------------------------------------------------------------------------------------------
# Setup CUDA and cuDNN if NGRAPH_GPU_ENABLE=TRUE
find_package(CUDA 8 QUIET)
if(CUDA_FOUND AND (NOT DEFINED NGRAPH_GPU_ENABLE OR NGRAPH_GPU_ENABLE))
if(CUDA_FOUND AND NGRAPH_GPU_ENABLE)
message(STATUS "GPU Backend Enabled")
set(NGRAPH_GPU_ENABLE TRUE)
find_package(CUDNN 5 QUIET REQUIRED)
......
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