Unverified Commit 05d80db3 authored by Matthew Brookhart's avatar Matthew Brookhart Committed by GitHub

Disable GPU by default (#463)

parent a01170fb
......@@ -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