Skip to content
This repository was archived by the owner on Mar 12, 2021. It is now read-only.

Commit 4742b8a

Browse files
committed
CEnum 0.3 compatibility.
1 parent f44cf19 commit 4742b8a

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
2727
[compat]
2828
AbstractFFTs = "0.4, 0.5"
2929
Adapt = "1.0"
30-
CEnum = "0.2"
30+
CEnum = "0.2, 0.3"
3131
CUDAapi = "3.0, 4.0"
3232
CUDAdrv = "6.0.1"
3333
CUDAnative = "3.0"

src/dnn/rnn.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ function RNNDesc{T}(mode::cudnnRNNMode_t, input::Int, hidden::Int; layers = 1) w
4747
inputMode = CUDNN_LINEAR_INPUT
4848
direction = CUDNN_UNIDIRECTIONAL
4949
algo = CUDNN_RNN_ALGO_STANDARD
50-
cudnnSetRNNDescriptor_v6(handle(),d[],hidden,layers,dropoutDesc,cudnnRNNInputMode_t(inputMode),cudnnDirectionMode_t(direction),mode,cudnnRNNAlgo_t(algo),cudnnDataType(T))
50+
cudnnSetRNNDescriptor_v6(handle(),d[],hidden,layers,dropoutDesc,inputMode,direction,mode,algo,cudnnDataType(T))
5151

5252
w =CuArrays.zeros(T, rnnParamSize(T, d[], input))
5353
# TODO: avoid reserve allocation here

0 commit comments

Comments
 (0)