From 2ff54ee0fd85b1641279cfcad041331986b34604 Mon Sep 17 00:00:00 2001 From: Tejan Karmali Date: Thu, 4 Oct 2018 11:31:29 -0400 Subject: [PATCH] cudnn_available() update --- src/cuda/cuda.jl | 2 +- test/cuda/cuda.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cuda/cuda.jl b/src/cuda/cuda.jl index fe36bf5d..15126aca 100644 --- a/src/cuda/cuda.jl +++ b/src/cuda/cuda.jl @@ -2,6 +2,6 @@ module CUDA using ..CuArrays -CuArrays.cudnn_available() && include("cudnn.jl") +CuArrays.libcudnn != nothing && include("cudnn.jl") end diff --git a/test/cuda/cuda.jl b/test/cuda/cuda.jl index 16f90e89..1f54d1b9 100644 --- a/test/cuda/cuda.jl +++ b/test/cuda/cuda.jl @@ -36,4 +36,4 @@ Flux.back!(sum(l)) end -CuArrays.cudnn_available() && include("cudnn.jl") +CuArrays.libcudnn != nothing && include("cudnn.jl")