diff --git a/Project.toml b/Project.toml index d17905d7..8efce4d7 100644 --- 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 index 32e15300..f9b7ecf6 100644 --- 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