TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix CuArrays.libcudnn imports

Browse Source
This commit is contained in:
janEbert 2019-08-24 02:21:54 +02:00
parent 5689b39538
commit 978d7bf195
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/cuda/cudnn.jl
View File

@ -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

3
src/cuda/curnn.jl
View File

@ -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