From de838925503901066c64294d688395960ca6b6a6 Mon Sep 17 00:00:00 2001 From: David Adrian Date: Fri, 9 Aug 2019 14:24:15 +0200 Subject: [PATCH 1/2] Add wrapper functions for setting/getting convolution group count --- src/dnn/libcudnn.jl | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/dnn/libcudnn.jl b/src/dnn/libcudnn.jl index a16ff41f..19d643e2 100644 --- a/src/dnn/libcudnn.jl +++ b/src/dnn/libcudnn.jl @@ -126,6 +126,20 @@ function cudnnSetConvolutionMathType(convDesc, mathType) convDesc, mathType) end +function cudnnSetConvolutionGroupCount(convDesc,groupCount) + @check ccall((:cudnnSetConvolutionGroupCount,libcudnn), + cudnnStatus_t, + (cudnnConvolutionDescriptor_t,Cint), + convDesc,groupCount) +end + +function cudnnGetConvolutionGroupCount(convDesc,groupCount) + @check ccall((:cudnnSetConvolutionGroupCount,libcudnn), + cudnnStatus_t, + (cudnnConvolutionDescriptor_t,Ptr{Cint}), + convDesc,groupCount) +end + function cudnnCreatePoolingDescriptor(poolingDesc) @check ccall((:cudnnCreatePoolingDescriptor,libcudnn), cudnnStatus_t, From a4665f4066ca38c0c56da52edf5196bbc730a3e0 Mon Sep 17 00:00:00 2001 From: "David B. Adrian" Date: Fri, 9 Aug 2019 19:05:42 +0200 Subject: [PATCH 2/2] Fix typo in function call --- src/dnn/libcudnn.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/dnn/libcudnn.jl b/src/dnn/libcudnn.jl index 19d643e2..8e2bb987 100644 --- a/src/dnn/libcudnn.jl +++ b/src/dnn/libcudnn.jl @@ -134,7 +134,7 @@ function cudnnSetConvolutionGroupCount(convDesc,groupCount) end function cudnnGetConvolutionGroupCount(convDesc,groupCount) - @check ccall((:cudnnSetConvolutionGroupCount,libcudnn), + @check ccall((:cudnnGetConvolutionGroupCount,libcudnn), cudnnStatus_t, (cudnnConvolutionDescriptor_t,Ptr{Cint}), convDesc,groupCount)