diff --git a/src/cuda/cudnn.jl b/src/cuda/cudnn.jl index 57de9f01..8bd8135e 100644 --- a/src/cuda/cudnn.jl +++ b/src/cuda/cudnn.jl @@ -1,4 +1,4 @@ -using CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, +using .CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, cudnnBatchNormMode_t, cudnnHandle_t, cudnnDataType, TensorDesc, FilterDesc import ..Flux: data using LinearAlgebra diff --git a/src/cuda/curnn.jl b/src/cuda/curnn.jl index e76437d7..a47947e0 100644 --- a/src/cuda/curnn.jl +++ b/src/cuda/curnn.jl @@ -1,4 +1,4 @@ -using CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, +using .CuArrays.CUDNN: @check, libcudnn, cudnnStatus_t, cudnnTensorDescriptor_t, cudnnBatchNormMode_t, cudnnHandle_t, cudnnDataType, TensorDesc, FilterDesc using LinearAlgebra