Skip to content

Commit

Permalink
Merge pull request #219 from okuta/remove-old-cudnn
Browse files Browse the repository at this point in the history
Remove old cuDNN support
  • Loading branch information
unnonouno committed Jul 14, 2017
2 parents 7617ec3 + bf0b3c1 commit 48e2296
Show file tree
Hide file tree
Showing 4 changed files with 76 additions and 359 deletions.
47 changes: 14 additions & 33 deletions cupy/cuda/cudnn.pxd
Expand Up @@ -176,9 +176,6 @@ cpdef setTensor4dDescriptorEx(size_t tensorDesc, int dataType,
cpdef setTensorNdDescriptor(size_t tensorDesc, int dataType, int nbDims,
size_t dimA, size_t strideA)
cpdef destroyTensorDescriptor(size_t tensorDesc)
cpdef addTensor_v2(
size_t handle, int mode, size_t alpha, size_t biasDesc,
size_t biasData, size_t beta, size_t srcDestDesc, size_t srcDestData)
cpdef addTensor_v3(size_t handle, size_t alpha, size_t bDesc,
size_t b, size_t beta, size_t yDesc, size_t y)

Expand All @@ -188,10 +185,6 @@ cpdef addTensor_v3(size_t handle, size_t alpha, size_t bDesc,
###############################################################################

cpdef size_t createFilterDescriptor() except *
cpdef setFilter4dDescriptor_v3(
size_t filterDesc, int dataType, int k, int c, int h, int w)
cpdef setFilterNdDescriptor_v3(
size_t filterDesc, int dataType, int nbDims, size_t filterDimA)
cpdef setFilter4dDescriptor_v4(
size_t filterDesc, int dataType, int format, int k, int c, int h, int w)
cpdef setFilterNdDescriptor_v4(
Expand All @@ -210,9 +203,6 @@ cpdef setConvolution2dDescriptor_v4(
cpdef setConvolution2dDescriptor_v5(
size_t convDesc, int pad_h, int pad_w, int u, int v, int dilation_h,
int dilation_w, int mode, size_t computeType)
cpdef setConvolutionNdDescriptor_v2(
size_t convDesc, int arrayLength, size_t padA, size_t filterStrideA,
size_t dilationA, int mode)
cpdef setConvolutionNdDescriptor_v3(
size_t convDesc, int arrayLength, size_t padA, size_t filterStrideA,
size_t dilationA, int mode, int dataType)
Expand Down Expand Up @@ -253,10 +243,6 @@ cpdef int getConvolutionBackwardFilterAlgorithm(
cpdef size_t getConvolutionBackwardFilterWorkspaceSize(
size_t handle, size_t srcDesc, size_t diffDesc, size_t convDesc,
size_t filterDesc, int algo) except *
cpdef convolutionBackwardFilter_v2(
size_t handle, size_t alpha, size_t srcDesc, size_t srcData,
size_t diffDesc, size_t diffData, size_t convDesc, size_t beta,
size_t gradDesc, size_t gradData)
cpdef convolutionBackwardFilter_v3(
size_t handle, size_t alpha, size_t srcDesc, size_t srcData,
size_t diffDesc, size_t diffData, size_t convDesc, int algo,
Expand All @@ -276,10 +262,6 @@ cpdef int getConvolutionBackwardDataAlgorithm(
cpdef size_t getConvolutionBackwardDataWorkspaceSize(
size_t handle, size_t filterDesc, size_t diffDesc, size_t convDesc,
size_t gradDesc, int algo) except *
cpdef convolutionBackwardData_v2(
size_t handle, size_t alpha, size_t filterDesc, size_t filterData,
size_t diffDesc, size_t diffData, size_t convDesc, size_t beta,
size_t gradDesc, size_t gradData)
cpdef convolutionBackwardData_v3(
size_t handle, size_t alpha, size_t filterDesc, size_t filterData,
size_t diffDesc, size_t diffData, size_t convDesc, int algo,
Expand All @@ -292,13 +274,13 @@ cpdef convolutionBackwardData_v3(
###############################################################################

cpdef size_t createPoolingDescriptor() except *
cpdef setPooling2dDescriptor_v3(
size_t poolingDesc, int mode, int windowHeight, int windowWidth,
int verticalPadding, int horizontalPadding, int verticalStride,
int horizontalStride)
cpdef setPoolingNdDescriptor_v3(
size_t poolingDesc, int mode, int nbDims, size_t windowDimA,
size_t paddingA, size_t strideA)
cpdef setPooling2dDescriptor_v4(
size_t poolingDesc, int mode, int maxpoolingNanOpt, int windowHeight,
int windowWidth, int verticalPadding, int horizontalPadding,
int verticalStride, int horizontalStride)
cpdef setPoolingNdDescriptor_v4(
size_t poolingDesc, int mode, int maxpoolingNanOpt, int nbDims,
size_t windowDimA, size_t paddingA, size_t strideA)
cpdef destroyPoolingDescriptor(size_t poolingDesc)
cpdef poolingForward(
size_t handle, size_t poolingDesc, size_t alpha, size_t srcDesc,
Expand Down Expand Up @@ -359,15 +341,14 @@ cpdef softmaxBackward(
size_t handle, int algorithm, int mode, size_t alpha, size_t srcDesc,
size_t srcData, size_t srcDiffDesc, size_t srcDiffData, size_t beta,
size_t destDiffDesc, size_t destDiffData)
cpdef activationForward_v3(
size_t handle, int mode, size_t alpha, size_t srcDesc, size_t srcData,
size_t beta, size_t dstDesc, size_t dstData)
cpdef activationBackward_v3(
size_t handle, int mode, size_t alpha, size_t srcDesc, size_t srcData,
size_t srcDiffDesc, size_t srcDiffData, size_t destDesc,
size_t destData, size_t beta, size_t destDiffDesc,
cpdef activationForward_v4(
size_t handle, size_t activationDesc, size_t alpha, size_t srcDesc,
size_t srcData, size_t beta, size_t dstDesc, size_t dstData)
cpdef activationBackward_v4(
size_t handle, size_t activationDesc, size_t alpha, size_t srcDesc,
size_t srcData, size_t srcDiffDesc, size_t srcDiffData,
size_t destDesc, size_t destData, size_t beta, size_t destDiffDesc,
size_t destDiffData)

cpdef size_t createDropoutDescriptor() except *
cpdef destroyDropoutDescriptor(size_t dropoutDesc)
cpdef size_t dropoutGetStatesSize(size_t handle) except *
Expand Down

0 comments on commit 48e2296

Please sign in to comment.