From 978d7bf1959f15ef303060d30c521668c34f39c4 Mon Sep 17 00:00:00 2001 From: janEbert Date: Sat, 24 Aug 2019 02:21:54 +0200 Subject: [PATCH] Fix CuArrays.libcudnn imports --- src/cuda/cudnn.jl | 3 ++- src/cuda/curnn.jl | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/cuda/cudnn.jl b/src/cuda/cudnn.jl index fac35a72..1a258173 100644 --- a/src/cuda/cudnn.jl +++ b/src/cuda/cudnn.jl @@ -1,5 +1,6 @@ -using .CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, +using .CuArrays.CUDNN: @check, cudnnStatus_t, cudnnTensorDescriptor_t, cudnnBatchNormMode_t, cudnnHandle_t, cudnnDataType, TensorDesc, FilterDesc +using .CuArrays: libcudnn import ..Flux: data using LinearAlgebra diff --git a/src/cuda/curnn.jl b/src/cuda/curnn.jl index 09f6d43c..fbe73e45 100644 --- a/src/cuda/curnn.jl +++ b/src/cuda/curnn.jl @@ -1,5 +1,6 @@ -using .CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, +using .CuArrays.CUDNN: @check, cudnnStatus_t, cudnnTensorDescriptor_t, cudnnBatchNormMode_t, cudnnHandle_t, cudnnDataType, TensorDesc, FilterDesc +using .CuArrays: libcudnn using LinearAlgebra const RNN_RELU = 0 # Stock RNN with ReLu activation