Skip to content

Commit 6fa6e7d

Browse files
authored
Missing Identity Activation Import
1 parent 45180b7 commit 6fa6e7d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ext/NNlibCUDA/src/cudnn/activations.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
using Base.Broadcast
55
using CUDA.CUDNN: cudnnActivationForward!, cudnnOpTensor!,
66
CUDNN_ACTIVATION_TANH,CUDNN_ACTIVATION_SIGMOID,CUDNN_ACTIVATION_ELU,
7-
CUDNN_ACTIVATION_RELU,CUDNN_ACTIVATION_CLIPPED_RELU,CUDNN_OP_TENSOR_MAX
7+
CUDNN_ACTIVATION_RELU,CUDNN_ACTIVATION_CLIPPED_RELU,CUDNN_OP_TENSOR_MAX,
8+
CUDNN_ACTIVATION_IDENTITY
89

910
for (f, op) in [
1011
CUDA.tanh => (src,dst)->cudnnActivationForward!(dst, src, mode=CUDNN_ACTIVATION_TANH),

0 commit comments

Comments
 (0)