Merge pull request #705 from JuliaGPU/tb/cenum

CEnum 0.3 compatibility.
JuliaGPU · May 5, 2020 · eac33f8 · eac33f8
2 parents f44cf19 + 4742b8a
commit eac33f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -27,7 +27,7 @@ TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 [compat]
 AbstractFFTs = "0.4, 0.5"
 Adapt = "1.0"
-CEnum = "0.2"
+CEnum = "0.2, 0.3"
 CUDAapi = "3.0, 4.0"
 CUDAdrv = "6.0.1"
 CUDAnative = "3.0"

diff --git a/src/dnn/rnn.jl b/src/dnn/rnn.jl
@@ -47,7 +47,7 @@ function RNNDesc{T}(mode::cudnnRNNMode_t, input::Int, hidden::Int; layers = 1) w
   inputMode = CUDNN_LINEAR_INPUT
   direction = CUDNN_UNIDIRECTIONAL
   algo = CUDNN_RNN_ALGO_STANDARD
-  cudnnSetRNNDescriptor_v6(handle(),d[],hidden,layers,dropoutDesc,cudnnRNNInputMode_t(inputMode),cudnnDirectionMode_t(direction),mode,cudnnRNNAlgo_t(algo),cudnnDataType(T))
+  cudnnSetRNNDescriptor_v6(handle(),d[],hidden,layers,dropoutDesc,inputMode,direction,mode,algo,cudnnDataType(T))
 
   w =CuArrays.zeros(T, rnnParamSize(T, d[], input))
   # TODO: avoid reserve allocation here