| // Auto-generated, do not edit. |
| |
| extern "C" { |
| |
| size_t CUDNNWINAPI cudnnGetVersion(void) { |
| using FuncPtr = size_t (CUDNNWINAPI *)(); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetVersion"); |
| if (!func_ptr) return 0; |
| return func_ptr(); |
| } |
| |
| size_t CUDNNWINAPI cudnnGetCudartVersion(void) { |
| using FuncPtr = size_t (CUDNNWINAPI *)(); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetCudartVersion"); |
| if (!func_ptr) return 0; |
| return func_ptr(); |
| } |
| |
| const char * CUDNNWINAPI cudnnGetErrorString(cudnnStatus_t status) { |
| using FuncPtr = const char * (CUDNNWINAPI *)(cudnnStatus_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetErrorString"); |
| if (!func_ptr) return "cudnnGetErrorString symbol not found."; |
| return func_ptr(status); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnQueryRuntimeError( |
| cudnnHandle_t handle, |
| cudnnStatus_t *rstatus, |
| cudnnErrQueryMode_t mode, |
| cudnnRuntimeTag_t *tag ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnStatus_t *, cudnnErrQueryMode_t, cudnnRuntimeTag_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnQueryRuntimeError"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rstatus, mode, tag); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetProperty(libraryPropertyType type, int *value) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(libraryPropertyType, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetProperty"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(type, value); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreate (cudnnHandle_t *handle) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreate"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroy (cudnnHandle_t handle) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroy"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetStream (cudnnHandle_t handle, cudaStream_t streamId) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudaStream_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetStream"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, streamId); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetStream (cudnnHandle_t handle, cudaStream_t *streamId) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudaStream_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetStream"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, streamId); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateTensorDescriptor( |
| cudnnTensorDescriptor_t *tensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetTensor4dDescriptor( |
| cudnnTensorDescriptor_t tensorDesc, |
| cudnnTensorFormat_t format, |
| cudnnDataType_t dataType, /* image data type */ |
| int n, /* number of inputs (batch size) */ |
| int c, /* number of input feature maps */ |
| int h, /* height of input section */ |
| int w ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, cudnnTensorFormat_t, cudnnDataType_t, int, int, int, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetTensor4dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, format, dataType, n, c, h, w); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetTensor4dDescriptorEx( |
| cudnnTensorDescriptor_t tensorDesc, |
| cudnnDataType_t dataType, /* image data type */ |
| int n, /* number of inputs (batch size) */ |
| int c, /* number of input feature maps */ |
| int h, /* height of input section */ |
| int w, /* width of input section */ |
| int nStride, |
| int cStride, |
| int hStride, |
| int wStride ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, cudnnDataType_t, int, int, int, int, int, int, int, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetTensor4dDescriptorEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, dataType, n, c, h, w, nStride, cStride, hStride, wStride); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetTensor4dDescriptor( |
| const cudnnTensorDescriptor_t tensorDesc, |
| cudnnDataType_t *dataType, /* image data type */ |
| int *n, /* number of inputs (batch size) */ |
| int *c, /* number of input feature maps */ |
| int *h, /* height of input section */ |
| int *w, /* width of input section */ |
| int *nStride, |
| int *cStride, |
| int *hStride, |
| int *wStride ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnTensorDescriptor_t, cudnnDataType_t *, int *, int *, int *, int *, int *, int *, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetTensor4dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, dataType, n, c, h, w, nStride, cStride, hStride, wStride); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetTensorNdDescriptor( |
| cudnnTensorDescriptor_t tensorDesc, |
| cudnnDataType_t dataType, |
| int nbDims, |
| const int dimA[], |
| const int strideA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, cudnnDataType_t, int, const int [], const int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetTensorNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, dataType, nbDims, dimA, strideA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetTensorNdDescriptorEx( |
| cudnnTensorDescriptor_t tensorDesc, |
| cudnnTensorFormat_t format, |
| cudnnDataType_t dataType, |
| int nbDims, |
| const int dimA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, cudnnTensorFormat_t, cudnnDataType_t, int, const int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetTensorNdDescriptorEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, format, dataType, nbDims, dimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetTensorNdDescriptor( |
| const cudnnTensorDescriptor_t tensorDesc, |
| int nbDimsRequested, |
| cudnnDataType_t *dataType, |
| int *nbDims, |
| int dimA[], |
| int strideA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnTensorDescriptor_t, int, cudnnDataType_t *, int *, int [], int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetTensorNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, nbDimsRequested, dataType, nbDims, dimA, strideA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetTensorSizeInBytes( |
| const cudnnTensorDescriptor_t tensorDesc, |
| size_t *size) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnTensorDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetTensorSizeInBytes"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc, size); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyTensorDescriptor( |
| cudnnTensorDescriptor_t tensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(tensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnTransformTensor( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnTransformTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, xDesc, x, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnAddTensor( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnTensorDescriptor_t aDesc, |
| const void *A, |
| const void *beta, |
| const cudnnTensorDescriptor_t cDesc, |
| void *C ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnAddTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, aDesc, A, beta, cDesc, C); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateOpTensorDescriptor( |
| cudnnOpTensorDescriptor_t *opTensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnOpTensorDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateOpTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(opTensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetOpTensorDescriptor( |
| cudnnOpTensorDescriptor_t opTensorDesc, |
| cudnnOpTensorOp_t opTensorOp, |
| cudnnDataType_t opTensorCompType, |
| cudnnNanPropagation_t opTensorNanOpt ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnOpTensorDescriptor_t, cudnnOpTensorOp_t, cudnnDataType_t, cudnnNanPropagation_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetOpTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(opTensorDesc, opTensorOp, opTensorCompType, opTensorNanOpt); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetOpTensorDescriptor( |
| const cudnnOpTensorDescriptor_t opTensorDesc, |
| cudnnOpTensorOp_t *opTensorOp, |
| cudnnDataType_t *opTensorCompType, |
| cudnnNanPropagation_t *opTensorNanOpt ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnOpTensorDescriptor_t, cudnnOpTensorOp_t *, cudnnDataType_t *, cudnnNanPropagation_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetOpTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(opTensorDesc, opTensorOp, opTensorCompType, opTensorNanOpt); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyOpTensorDescriptor( |
| cudnnOpTensorDescriptor_t opTensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnOpTensorDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyOpTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(opTensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnOpTensor( |
| cudnnHandle_t handle, |
| const cudnnOpTensorDescriptor_t opTensorDesc, |
| const void *alpha1, |
| const cudnnTensorDescriptor_t aDesc, |
| const void *A, |
| const void *alpha2, |
| const cudnnTensorDescriptor_t bDesc, |
| const void *B, |
| const void *beta, |
| const cudnnTensorDescriptor_t cDesc, |
| void *C ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnOpTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnOpTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, opTensorDesc, alpha1, aDesc, A, alpha2, bDesc, B, beta, cDesc, C); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateReduceTensorDescriptor( |
| cudnnReduceTensorDescriptor_t *reduceTensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnReduceTensorDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateReduceTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(reduceTensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetReduceTensorDescriptor( |
| cudnnReduceTensorDescriptor_t reduceTensorDesc, |
| cudnnReduceTensorOp_t reduceTensorOp, |
| cudnnDataType_t reduceTensorCompType, |
| cudnnNanPropagation_t reduceTensorNanOpt, |
| cudnnReduceTensorIndices_t reduceTensorIndices, |
| cudnnIndicesType_t reduceTensorIndicesType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnReduceTensorDescriptor_t, cudnnReduceTensorOp_t, cudnnDataType_t, cudnnNanPropagation_t, cudnnReduceTensorIndices_t, cudnnIndicesType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetReduceTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(reduceTensorDesc, reduceTensorOp, reduceTensorCompType, reduceTensorNanOpt, reduceTensorIndices, reduceTensorIndicesType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetReduceTensorDescriptor( |
| const cudnnReduceTensorDescriptor_t reduceTensorDesc, |
| cudnnReduceTensorOp_t *reduceTensorOp, |
| cudnnDataType_t *reduceTensorCompType, |
| cudnnNanPropagation_t *reduceTensorNanOpt, |
| cudnnReduceTensorIndices_t *reduceTensorIndices, |
| cudnnIndicesType_t *reduceTensorIndicesType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnReduceTensorDescriptor_t, cudnnReduceTensorOp_t *, cudnnDataType_t *, cudnnNanPropagation_t *, cudnnReduceTensorIndices_t *, cudnnIndicesType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetReduceTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(reduceTensorDesc, reduceTensorOp, reduceTensorCompType, reduceTensorNanOpt, reduceTensorIndices, reduceTensorIndicesType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyReduceTensorDescriptor( |
| cudnnReduceTensorDescriptor_t reduceTensorDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnReduceTensorDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyReduceTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(reduceTensorDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetReductionIndicesSize( |
| cudnnHandle_t handle, |
| const cudnnReduceTensorDescriptor_t reduceTensorDesc, |
| const cudnnTensorDescriptor_t aDesc, |
| const cudnnTensorDescriptor_t cDesc, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnReduceTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetReductionIndicesSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, reduceTensorDesc, aDesc, cDesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetReductionWorkspaceSize( |
| cudnnHandle_t handle, |
| const cudnnReduceTensorDescriptor_t reduceTensorDesc, |
| const cudnnTensorDescriptor_t aDesc, |
| const cudnnTensorDescriptor_t cDesc, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnReduceTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetReductionWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, reduceTensorDesc, aDesc, cDesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnReduceTensor( |
| cudnnHandle_t handle, |
| const cudnnReduceTensorDescriptor_t reduceTensorDesc, |
| void *indices, |
| size_t indicesSizeInBytes, |
| void *workspace, |
| size_t workspaceSizeInBytes, |
| const void *alpha, |
| const cudnnTensorDescriptor_t aDesc, |
| const void *A, |
| const void *beta, |
| const cudnnTensorDescriptor_t cDesc, |
| void *C ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnReduceTensorDescriptor_t, void *, size_t, void *, size_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnReduceTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, reduceTensorDesc, indices, indicesSizeInBytes, workspace, workspaceSizeInBytes, alpha, aDesc, A, beta, cDesc, C); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetTensor( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y, |
| const void *valuePtr ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, void *, const void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, yDesc, y, valuePtr); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnScaleTensor( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y, |
| const void *alpha ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, void *, const void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnScaleTensor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, yDesc, y, alpha); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateFilterDescriptor( |
| cudnnFilterDescriptor_t *filterDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnFilterDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateFilterDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetFilter4dDescriptor( |
| cudnnFilterDescriptor_t filterDesc, |
| cudnnDataType_t dataType, /* image data type */ |
| cudnnTensorFormat_t format, |
| int k, /* number of output feature maps */ |
| int c, /* number of input feature maps */ |
| int h, /* height of each input filter */ |
| int w ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnFilterDescriptor_t, cudnnDataType_t, cudnnTensorFormat_t, int, int, int, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetFilter4dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc, dataType, format, k, c, h, w); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetFilter4dDescriptor( |
| const cudnnFilterDescriptor_t filterDesc, |
| cudnnDataType_t *dataType, /* image data type */ |
| cudnnTensorFormat_t *format, |
| int *k, /* number of output feature maps */ |
| int *c, /* number of input feature maps */ |
| int *h, /* height of each input filter */ |
| int *w ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnFilterDescriptor_t, cudnnDataType_t *, cudnnTensorFormat_t *, int *, int *, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetFilter4dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc, dataType, format, k, c, h, w); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetFilterNdDescriptor( |
| cudnnFilterDescriptor_t filterDesc, |
| cudnnDataType_t dataType, /* image data type */ |
| cudnnTensorFormat_t format, |
| int nbDims, |
| const int filterDimA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnFilterDescriptor_t, cudnnDataType_t, cudnnTensorFormat_t, int, const int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetFilterNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc, dataType, format, nbDims, filterDimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetFilterNdDescriptor( |
| const cudnnFilterDescriptor_t filterDesc, |
| int nbDimsRequested, |
| cudnnDataType_t *dataType, /* image data type */ |
| cudnnTensorFormat_t *format, |
| int *nbDims, |
| int filterDimA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnFilterDescriptor_t, int, cudnnDataType_t *, cudnnTensorFormat_t *, int *, int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetFilterNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc, nbDimsRequested, dataType, format, nbDims, filterDimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyFilterDescriptor( |
| cudnnFilterDescriptor_t filterDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnFilterDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyFilterDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(filterDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateConvolutionDescriptor( |
| cudnnConvolutionDescriptor_t *convDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateConvolutionDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetConvolutionMathType( cudnnConvolutionDescriptor_t convDesc, |
| cudnnMathType_t mathType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, cudnnMathType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetConvolutionMathType"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, mathType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionMathType( cudnnConvolutionDescriptor_t convDesc, |
| cudnnMathType_t *mathType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, cudnnMathType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionMathType"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, mathType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetConvolutionGroupCount( cudnnConvolutionDescriptor_t convDesc, |
| int groupCount ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetConvolutionGroupCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, groupCount); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionGroupCount( cudnnConvolutionDescriptor_t convDesc, |
| int *groupCount ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionGroupCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, groupCount); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetConvolution2dDescriptor( cudnnConvolutionDescriptor_t convDesc, |
| int pad_h, /* zero-padding height */ |
| int pad_w, /* zero-padding width */ |
| int u, /* vertical filter stride */ |
| int v, /* horizontal filter stride */ |
| int dilation_h, /* filter dilation in the vertical dimension */ |
| int dilation_w, /* filter dilation in the horizontal dimension */ |
| cudnnConvolutionMode_t mode, |
| cudnnDataType_t computeType |
| ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, int, int, int, int, int, int, cudnnConvolutionMode_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetConvolution2dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolution2dDescriptor( const cudnnConvolutionDescriptor_t convDesc, |
| int* pad_h, /* zero-padding height */ |
| int* pad_w, /* zero-padding width */ |
| int* u, /* vertical filter stride */ |
| int* v, /* horizontal filter stride */ |
| int* dilation_h, /* filter dilation in the vertical dimension */ |
| int* dilation_w, /* filter dilation in the horizontal dimension */ |
| cudnnConvolutionMode_t* mode, |
| cudnnDataType_t *computeType |
| ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnConvolutionDescriptor_t, int *, int *, int *, int *, int *, int *, cudnnConvolutionMode_t *, cudnnDataType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolution2dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolution2dForwardOutputDim( |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t inputTensorDesc, |
| const cudnnFilterDescriptor_t filterDesc, |
| int *n, |
| int *c, |
| int *h, |
| int *w ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, int *, int *, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolution2dForwardOutputDim"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, inputTensorDesc, filterDesc, n, c, h, w); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetConvolutionNdDescriptor( |
| cudnnConvolutionDescriptor_t convDesc, |
| int arrayLength, /* nbDims-2 size */ |
| const int padA[], |
| const int filterStrideA[], |
| const int dilationA[], |
| cudnnConvolutionMode_t mode, |
| cudnnDataType_t computeType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, int, const int [], const int [], const int [], cudnnConvolutionMode_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetConvolutionNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, arrayLength, padA, filterStrideA, dilationA, mode, computeType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionNdDescriptor( |
| const cudnnConvolutionDescriptor_t convDesc, |
| int arrayLengthRequested, |
| int *arrayLength, |
| int padA[], |
| int strideA[], |
| int dilationA[], |
| cudnnConvolutionMode_t *mode, |
| cudnnDataType_t *computeType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnConvolutionDescriptor_t, int, int *, int [], int [], int [], cudnnConvolutionMode_t *, cudnnDataType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, arrayLengthRequested, arrayLength, padA, strideA, dilationA, mode, computeType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionNdForwardOutputDim( |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t inputTensorDesc, |
| const cudnnFilterDescriptor_t filterDesc, |
| int nbDims, |
| int tensorOutputDimA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, int, int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionNdForwardOutputDim"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc, inputTensorDesc, filterDesc, nbDims, tensorOutputDimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyConvolutionDescriptor( |
| cudnnConvolutionDescriptor_t convDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyConvolutionDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(convDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionForwardAlgorithmMaxCount( cudnnHandle_t handle, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionForwardAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionForwardAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t yDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionFwdAlgoPerf_t *perfResults ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const int, int *, cudnnConvolutionFwdAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionForwardAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionForwardAlgorithmEx( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionFwdAlgoPerf_t *perfResults, |
| void *workSpace, |
| size_t workSpaceSizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, void *, const int, int *, cudnnConvolutionFwdAlgoPerf_t *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionForwardAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, x, wDesc, w, convDesc, yDesc, y, requestedAlgoCount, returnedAlgoCount, perfResults, workSpace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionForwardAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t yDesc, |
| cudnnConvolutionFwdPreference_t preference, |
| size_t memoryLimitInBytes, |
| cudnnConvolutionFwdAlgo_t *algo ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, cudnnConvolutionFwdPreference_t, size_t, cudnnConvolutionFwdAlgo_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionForwardAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, preference, memoryLimitInBytes, algo); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionForwardAlgorithm_v7( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t srcDesc, |
| const cudnnFilterDescriptor_t filterDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t destDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionFwdAlgoPerf_t *perfResults) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const int, int *, cudnnConvolutionFwdAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionForwardAlgorithm_v7"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, srcDesc, filterDesc, convDesc, destDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionForwardWorkspaceSize( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t yDesc, |
| cudnnConvolutionFwdAlgo_t algo, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, cudnnConvolutionFwdAlgo_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionForwardWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, wDesc, convDesc, yDesc, algo, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnConvolutionForward( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnConvolutionDescriptor_t convDesc, |
| cudnnConvolutionFwdAlgo_t algo, |
| void *workSpace, |
| size_t workSpaceSizeInBytes, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, cudnnConvolutionFwdAlgo_t, void *, size_t, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnConvolutionForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, xDesc, x, wDesc, w, convDesc, algo, workSpace, workSpaceSizeInBytes, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnConvolutionBiasActivationForward( |
| cudnnHandle_t handle, |
| const void *alpha1, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnConvolutionDescriptor_t convDesc, |
| cudnnConvolutionFwdAlgo_t algo, |
| void *workSpace, |
| size_t workSpaceSizeInBytes, |
| const void *alpha2, |
| const cudnnTensorDescriptor_t zDesc, |
| const void *z, |
| const cudnnTensorDescriptor_t biasDesc, |
| const void *bias, |
| const cudnnActivationDescriptor_t activationDesc, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, cudnnConvolutionFwdAlgo_t, void *, size_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnActivationDescriptor_t, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnConvolutionBiasActivationForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha1, xDesc, x, wDesc, w, convDesc, algo, workSpace, workSpaceSizeInBytes, alpha2, zDesc, z, biasDesc, bias, activationDesc, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnConvolutionBackwardBias( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const void *beta, |
| const cudnnTensorDescriptor_t dbDesc, |
| void *db ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnConvolutionBackwardBias"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, dyDesc, dy, beta, dbDesc, db); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardFilterAlgorithmMaxCount( cudnnHandle_t handle, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardFilterAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionBackwardFilterAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnFilterDescriptor_t dwDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdFilterAlgoPerf_t *perfResults ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnFilterDescriptor_t, const int, int *, cudnnConvolutionBwdFilterAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionBackwardFilterAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionBackwardFilterAlgorithmEx( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *y, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnFilterDescriptor_t dwDesc, |
| void *dw, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdFilterAlgoPerf_t *perfResults, |
| void *workSpace, |
| size_t workSpaceSizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, const cudnnFilterDescriptor_t, void *, const int, int *, cudnnConvolutionBwdFilterAlgoPerf_t *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionBackwardFilterAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, x, dyDesc, y, convDesc, dwDesc, dw, requestedAlgoCount, returnedAlgoCount, perfResults, workSpace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardFilterAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnFilterDescriptor_t dwDesc, |
| cudnnConvolutionBwdFilterPreference_t preference, |
| size_t memoryLimitInBytes, |
| cudnnConvolutionBwdFilterAlgo_t *algo ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnFilterDescriptor_t, cudnnConvolutionBwdFilterPreference_t, size_t, cudnnConvolutionBwdFilterAlgo_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardFilterAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, dyDesc, convDesc, dwDesc, preference, memoryLimitInBytes, algo); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardFilterAlgorithm_v7( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t srcDesc, |
| const cudnnTensorDescriptor_t diffDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnFilterDescriptor_t gradDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdFilterAlgoPerf_t *perfResults) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnFilterDescriptor_t, const int, int *, cudnnConvolutionBwdFilterAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardFilterAlgorithm_v7"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, srcDesc, diffDesc, convDesc, gradDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardFilterWorkspaceSize( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnFilterDescriptor_t gradDesc, |
| cudnnConvolutionBwdFilterAlgo_t algo, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnFilterDescriptor_t, cudnnConvolutionBwdFilterAlgo_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardFilterWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, dyDesc, convDesc, gradDesc, algo, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnConvolutionBackwardFilter( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnConvolutionDescriptor_t convDesc, |
| cudnnConvolutionBwdFilterAlgo_t algo, |
| void *workSpace, |
| size_t workSpaceSizeInBytes, |
| const void *beta, |
| const cudnnFilterDescriptor_t dwDesc, |
| void *dw ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, cudnnConvolutionBwdFilterAlgo_t, void *, size_t, const void *, const cudnnFilterDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnConvolutionBackwardFilter"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, xDesc, x, dyDesc, dy, convDesc, algo, workSpace, workSpaceSizeInBytes, beta, dwDesc, dw); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardDataAlgorithmMaxCount( cudnnHandle_t handle, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardDataAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionBackwardDataAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t dxDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdDataAlgoPerf_t *perfResults ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnFilterDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const int, int *, cudnnConvolutionBwdDataAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionBackwardDataAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, wDesc, dyDesc, convDesc, dxDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindConvolutionBackwardDataAlgorithmEx( |
| cudnnHandle_t handle, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdDataAlgoPerf_t *perfResults, |
| void *workSpace, |
| size_t workSpaceSizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, void *, const int, int *, cudnnConvolutionBwdDataAlgoPerf_t *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindConvolutionBackwardDataAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, wDesc, w, dyDesc, dy, convDesc, dxDesc, dx, requestedAlgoCount, returnedAlgoCount, perfResults, workSpace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardDataAlgorithm( |
| cudnnHandle_t handle, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t dxDesc, |
| cudnnConvolutionBwdDataPreference_t preference, |
| size_t memoryLimitInBytes, |
| cudnnConvolutionBwdDataAlgo_t *algo ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnFilterDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, cudnnConvolutionBwdDataPreference_t, size_t, cudnnConvolutionBwdDataAlgo_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardDataAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, wDesc, dyDesc, convDesc, dxDesc, preference, memoryLimitInBytes, algo); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardDataAlgorithm_v7( |
| cudnnHandle_t handle, |
| const cudnnFilterDescriptor_t filterDesc, |
| const cudnnTensorDescriptor_t diffDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t gradDesc, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnConvolutionBwdDataAlgoPerf_t *perfResults) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnFilterDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, const int, int *, cudnnConvolutionBwdDataAlgoPerf_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardDataAlgorithm_v7"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, filterDesc, diffDesc, convDesc, gradDesc, requestedAlgoCount, returnedAlgoCount, perfResults); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetConvolutionBackwardDataWorkspaceSize( |
| cudnnHandle_t handle, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnTensorDescriptor_t dyDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| const cudnnTensorDescriptor_t dxDesc, |
| cudnnConvolutionBwdDataAlgo_t algo, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnFilterDescriptor_t, const cudnnTensorDescriptor_t, const cudnnConvolutionDescriptor_t, const cudnnTensorDescriptor_t, cudnnConvolutionBwdDataAlgo_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolutionBackwardDataWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, wDesc, dyDesc, convDesc, dxDesc, algo, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnConvolutionBackwardData( |
| cudnnHandle_t handle, |
| const void *alpha, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnConvolutionDescriptor_t convDesc, |
| cudnnConvolutionBwdDataAlgo_t algo, |
| void *workSpace, |
| size_t workSpaceSizeInBytes, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnConvolutionDescriptor_t, cudnnConvolutionBwdDataAlgo_t, void *, size_t, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnConvolutionBackwardData"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, alpha, wDesc, w, dyDesc, dy, convDesc, algo, workSpace, workSpaceSizeInBytes, beta, dxDesc, dx); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnIm2Col( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const cudnnFilterDescriptor_t wDesc, |
| const cudnnConvolutionDescriptor_t convDesc, |
| void *colBuffer ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const cudnnConvolutionDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnIm2Col"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, xDesc, x, wDesc, convDesc, colBuffer); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSoftmaxForward( |
| cudnnHandle_t handle, |
| cudnnSoftmaxAlgorithm_t algo, |
| cudnnSoftmaxMode_t mode, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnSoftmaxAlgorithm_t, cudnnSoftmaxMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSoftmaxForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, algo, mode, alpha, xDesc, x, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSoftmaxBackward( |
| cudnnHandle_t handle, |
| cudnnSoftmaxAlgorithm_t algo, |
| cudnnSoftmaxMode_t mode, |
| const void *alpha, |
| const cudnnTensorDescriptor_t yDesc, |
| const void *y, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnSoftmaxAlgorithm_t, cudnnSoftmaxMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSoftmaxBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, algo, mode, alpha, yDesc, y, dyDesc, dy, beta, dxDesc, dx); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreatePoolingDescriptor( |
| cudnnPoolingDescriptor_t *poolingDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnPoolingDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreatePoolingDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetPooling2dDescriptor( |
| cudnnPoolingDescriptor_t poolingDesc, |
| cudnnPoolingMode_t mode, |
| cudnnNanPropagation_t maxpoolingNanOpt, |
| int windowHeight, |
| int windowWidth, |
| int verticalPadding, |
| int horizontalPadding, |
| int verticalStride, |
| int horizontalStride ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnPoolingDescriptor_t, cudnnPoolingMode_t, cudnnNanPropagation_t, int, int, int, int, int, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetPooling2dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, mode, maxpoolingNanOpt, windowHeight, windowWidth, verticalPadding, horizontalPadding, verticalStride, horizontalStride); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetPooling2dDescriptor( |
| const cudnnPoolingDescriptor_t poolingDesc, |
| cudnnPoolingMode_t *mode, |
| cudnnNanPropagation_t *maxpoolingNanOpt, |
| int *windowHeight, |
| int *windowWidth, |
| int *verticalPadding, |
| int *horizontalPadding, |
| int *verticalStride, |
| int *horizontalStride ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnPoolingDescriptor_t, cudnnPoolingMode_t *, cudnnNanPropagation_t *, int *, int *, int *, int *, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetPooling2dDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, mode, maxpoolingNanOpt, windowHeight, windowWidth, verticalPadding, horizontalPadding, verticalStride, horizontalStride); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetPoolingNdDescriptor( |
| cudnnPoolingDescriptor_t poolingDesc, |
| const cudnnPoolingMode_t mode, |
| const cudnnNanPropagation_t maxpoolingNanOpt, |
| int nbDims, |
| const int windowDimA[], |
| const int paddingA[], |
| const int strideA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnPoolingDescriptor_t, const cudnnPoolingMode_t, const cudnnNanPropagation_t, int, const int [], const int [], const int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetPoolingNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, mode, maxpoolingNanOpt, nbDims, windowDimA, paddingA, strideA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetPoolingNdDescriptor( |
| const cudnnPoolingDescriptor_t poolingDesc, |
| int nbDimsRequested, |
| cudnnPoolingMode_t *mode, |
| cudnnNanPropagation_t *maxpoolingNanOpt, |
| int *nbDims, |
| int windowDimA[], |
| int paddingA[], |
| int strideA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnPoolingDescriptor_t, int, cudnnPoolingMode_t *, cudnnNanPropagation_t *, int *, int [], int [], int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetPoolingNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, nbDimsRequested, mode, maxpoolingNanOpt, nbDims, windowDimA, paddingA, strideA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetPoolingNdForwardOutputDim( |
| const cudnnPoolingDescriptor_t poolingDesc, |
| const cudnnTensorDescriptor_t inputTensorDesc, |
| int nbDims, |
| int outputTensorDimA[] ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnPoolingDescriptor_t, const cudnnTensorDescriptor_t, int, int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetPoolingNdForwardOutputDim"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, inputTensorDesc, nbDims, outputTensorDimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetPooling2dForwardOutputDim( |
| const cudnnPoolingDescriptor_t poolingDesc, |
| const cudnnTensorDescriptor_t inputTensorDesc, |
| int *n, |
| int *c, |
| int *h, |
| int *w ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnPoolingDescriptor_t, const cudnnTensorDescriptor_t, int *, int *, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetPooling2dForwardOutputDim"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc, inputTensorDesc, n, c, h, w); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyPoolingDescriptor( |
| cudnnPoolingDescriptor_t poolingDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnPoolingDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyPoolingDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(poolingDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnPoolingForward( |
| cudnnHandle_t handle, |
| const cudnnPoolingDescriptor_t poolingDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnPoolingDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnPoolingForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, poolingDesc, alpha, xDesc, x, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnPoolingBackward( |
| cudnnHandle_t handle, |
| const cudnnPoolingDescriptor_t poolingDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t yDesc, |
| const void *y, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnPoolingDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnPoolingBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, poolingDesc, alpha, yDesc, y, dyDesc, dy, xDesc, x, beta, dxDesc, dx); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateActivationDescriptor( |
| cudnnActivationDescriptor_t *activationDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnActivationDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateActivationDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(activationDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetActivationDescriptor( |
| cudnnActivationDescriptor_t activationDesc, |
| cudnnActivationMode_t mode, |
| cudnnNanPropagation_t reluNanOpt, |
| double coef ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnActivationDescriptor_t, cudnnActivationMode_t, cudnnNanPropagation_t, double); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetActivationDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(activationDesc, mode, reluNanOpt, coef); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetActivationDescriptor( |
| const cudnnActivationDescriptor_t activationDesc, |
| cudnnActivationMode_t *mode, |
| cudnnNanPropagation_t *reluNanOpt, |
| double* coef ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnActivationDescriptor_t, cudnnActivationMode_t *, cudnnNanPropagation_t *, double *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetActivationDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(activationDesc, mode, reluNanOpt, coef); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyActivationDescriptor( |
| cudnnActivationDescriptor_t activationDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnActivationDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyActivationDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(activationDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnActivationForward( |
| cudnnHandle_t handle, |
| cudnnActivationDescriptor_t activationDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnActivationDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnActivationForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, activationDesc, alpha, xDesc, x, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnActivationBackward( |
| cudnnHandle_t handle, |
| cudnnActivationDescriptor_t activationDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t yDesc, |
| const void *y, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnActivationDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnActivationBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, activationDesc, alpha, yDesc, y, dyDesc, dy, xDesc, x, beta, dxDesc, dx); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateLRNDescriptor( |
| cudnnLRNDescriptor_t *normDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnLRNDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateLRNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(normDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetLRNDescriptor( |
| cudnnLRNDescriptor_t normDesc, |
| unsigned lrnN, |
| double lrnAlpha, |
| double lrnBeta, |
| double lrnK ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnLRNDescriptor_t, unsigned int, double, double, double); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetLRNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetLRNDescriptor( |
| cudnnLRNDescriptor_t normDesc, |
| unsigned* lrnN, |
| double* lrnAlpha, |
| double* lrnBeta, |
| double* lrnK ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnLRNDescriptor_t, unsigned int *, double *, double *, double *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetLRNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(normDesc, lrnN, lrnAlpha, lrnBeta, lrnK); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyLRNDescriptor( cudnnLRNDescriptor_t lrnDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnLRNDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyLRNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(lrnDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnLRNCrossChannelForward( |
| cudnnHandle_t handle, |
| cudnnLRNDescriptor_t normDesc, |
| cudnnLRNMode_t lrnMode, |
| const void* alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnLRNDescriptor_t, cudnnLRNMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnLRNCrossChannelForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, normDesc, lrnMode, alpha, xDesc, x, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnLRNCrossChannelBackward( |
| cudnnHandle_t handle, |
| cudnnLRNDescriptor_t normDesc, |
| cudnnLRNMode_t lrnMode, |
| const void* alpha, |
| const cudnnTensorDescriptor_t yDesc, |
| const void *y, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnLRNDescriptor_t, cudnnLRNMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnLRNCrossChannelBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, normDesc, lrnMode, alpha, yDesc, y, dyDesc, dy, xDesc, x, beta, dxDesc, dx); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDivisiveNormalizationForward( |
| cudnnHandle_t handle, |
| cudnnLRNDescriptor_t normDesc, |
| cudnnDivNormMode_t mode, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, /* same desc for means, temp, temp2 */ |
| const void *x, |
| const void *means, /* if NULL, means are assumed to be zero */ |
| void *temp, |
| void *temp2, |
| const void *beta, |
| const cudnnTensorDescriptor_t yDesc, |
| void *y ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnLRNDescriptor_t, cudnnDivNormMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, void *, void *, const void *, const cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDivisiveNormalizationForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, normDesc, mode, alpha, xDesc, x, means, temp, temp2, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDivisiveNormalizationBackward( |
| cudnnHandle_t handle, |
| cudnnLRNDescriptor_t normDesc, |
| cudnnDivNormMode_t mode, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, /* same desc for x, means, dy, temp, temp2 */ |
| const void *x, |
| const void *means, /* if NULL, means are assumed to be zero */ |
| const void *dy, |
| void *temp, |
| void *temp2, |
| const void *beta, |
| const cudnnTensorDescriptor_t dXdMeansDesc, /* same desc for dx, dMeans */ |
| void *dx, /* output x differential */ |
| void *dMeans ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnLRNDescriptor_t, cudnnDivNormMode_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const void *, void *, void *, const void *, const cudnnTensorDescriptor_t, void *, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDivisiveNormalizationBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, normDesc, mode, alpha, xDesc, x, means, dy, temp, temp2, beta, dXdMeansDesc, dx, dMeans); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDeriveBNTensorDescriptor( |
| cudnnTensorDescriptor_t derivedBnDesc, |
| const cudnnTensorDescriptor_t xDesc, |
| cudnnBatchNormMode_t mode ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, cudnnBatchNormMode_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDeriveBNTensorDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(derivedBnDesc, xDesc, mode); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnBatchNormalizationForwardTraining( |
| cudnnHandle_t handle, |
| cudnnBatchNormMode_t mode, |
| |
| const void *alpha, /* alpha[0] = result blend factor */ |
| const void *beta, /* beta[0] = dest layer blend factor */ |
| |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, /* NxCxHxW */ |
| const cudnnTensorDescriptor_t yDesc, |
| void *y, /* NxCxHxW */ |
| |
| /* Shared desc for the next 6 tensors in the argument list. |
| Data type to be set as follows: |
| type = (typeOf(x) == double) ? double : float |
| Dimensions for this descriptor depend on normalization mode |
| - Spatial Normalization : tensors are expected to have dims 1xCx1x1 |
| (normalization is performed across NxHxW) |
| - Per-Activation Normalization : tensors are expected to have dims of 1xCxHxW |
| (normalization is performed across N) */ |
| const cudnnTensorDescriptor_t bnScaleBiasMeanVarDesc, |
| |
| /* 'Gamma' and 'Beta' respectively in Ioffe and Szegedy's paper's notation */ |
| const void *bnScale, |
| const void *bnBias, |
| |
| /* MUST use factor=1 in the very first call of a complete training cycle. |
| Use a factor=1/(1+n) at N-th call to the function to get |
| Cumulative Moving Average (CMA) behavior |
| CMA[n] = (x[1]+...+x[n])/n |
| Since CMA[n+1] = (n*CMA[n]+x[n+1])/(n+1) = |
| ((n+1)*CMA[n]-CMA[n])/(n+1) + x[n+1]/(n+1) = |
| CMA[n]*(1-1/(n+1)) + x[n+1]*1/(n+1) */ |
| double exponentialAverageFactor, |
| |
| /* Used in Training phase only. |
| runningMean = newMean*factor + runningMean*(1-factor) */ |
| void *resultRunningMean, |
| /* Output in training mode, input in inference. Is the moving average |
| of variance[x] (factor is applied in the same way as for runningMean) */ |
| void *resultRunningVariance, |
| |
| /* Has to be >= CUDNN_BN_MIN_EPSILON. Should be the same in forward and backward functions. */ |
| double epsilon, |
| |
| /* Optionally save intermediate results from the forward pass here |
| - can be reused to speed up backward pass. NULL if unused */ |
| void *resultSaveMean, |
| void *resultSaveInvVariance ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnBatchNormMode_t, const void *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, const void *, const void *, double, void *, void *, double, void *, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnBatchNormalizationForwardTraining"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, mode, alpha, beta, xDesc, x, yDesc, y, bnScaleBiasMeanVarDesc, bnScale, bnBias, exponentialAverageFactor, resultRunningMean, resultRunningVariance, epsilon, resultSaveMean, resultSaveInvVariance); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnBatchNormalizationForwardInference( |
| cudnnHandle_t handle, |
| cudnnBatchNormMode_t mode, |
| const void *alpha, /* alpha[0] = result blend factor */ |
| const void *beta, /* beta[0] = dest layer blend factor */ |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, /* NxCxHxW */ |
| const cudnnTensorDescriptor_t yDesc, |
| void *y, /* NxCxHxW */ |
| const cudnnTensorDescriptor_t bnScaleBiasMeanVarDesc, |
| const void *bnScale, |
| const void *bnBias, |
| const void *estimatedMean, |
| const void *estimatedVariance, |
| double epsilon ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnBatchNormMode_t, const void *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, const void *, const void *, const void *, const void *, double); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnBatchNormalizationForwardInference"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, mode, alpha, beta, xDesc, x, yDesc, y, bnScaleBiasMeanVarDesc, bnScale, bnBias, estimatedMean, estimatedVariance, epsilon); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnBatchNormalizationBackward( |
| cudnnHandle_t handle, |
| cudnnBatchNormMode_t mode, |
| const void *alphaDataDiff, |
| const void *betaDataDiff, |
| const void *alphaParamDiff, |
| const void *betaParamDiff, |
| const cudnnTensorDescriptor_t xDesc, /* same desc for x, dx, dy */ |
| const void *x, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx, |
| /* Shared tensor desc for the 4 tensors below */ |
| const cudnnTensorDescriptor_t dBnScaleBiasDesc, |
| const void *bnScale, /* bnBias doesn't affect backpropagation */ |
| /* scale and bias diff are not backpropagated below this layer */ |
| void *dBnScaleResult, |
| void *dBnBiasResult, |
| /* Same epsilon as forward pass */ |
| double epsilon, |
| |
| /* Optionally cached intermediate results from |
| forward pass */ |
| const void *savedMean, |
| const void *savedInvVariance ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnBatchNormMode_t, const void *, const void *, const void *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, const void *, void *, void *, double, const void *, const void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnBatchNormalizationBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, mode, alphaDataDiff, betaDataDiff, alphaParamDiff, betaParamDiff, xDesc, x, dyDesc, dy, dxDesc, dx, dBnScaleBiasDesc, bnScale, dBnScaleResult, dBnBiasResult, epsilon, savedMean, savedInvVariance); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateSpatialTransformerDescriptor( |
| cudnnSpatialTransformerDescriptor_t *stDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnSpatialTransformerDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateSpatialTransformerDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(stDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetSpatialTransformerNdDescriptor( |
| cudnnSpatialTransformerDescriptor_t stDesc, |
| cudnnSamplerType_t samplerType, |
| cudnnDataType_t dataType, |
| const int nbDims, |
| const int dimA[]) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnSpatialTransformerDescriptor_t, cudnnSamplerType_t, cudnnDataType_t, const int, const int []); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetSpatialTransformerNdDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(stDesc, samplerType, dataType, nbDims, dimA); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroySpatialTransformerDescriptor( |
| cudnnSpatialTransformerDescriptor_t stDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnSpatialTransformerDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroySpatialTransformerDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(stDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSpatialTfGridGeneratorForward( |
| cudnnHandle_t handle, |
| const cudnnSpatialTransformerDescriptor_t stDesc, |
| const void *theta, |
| void *grid) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnSpatialTransformerDescriptor_t, const void *, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSpatialTfGridGeneratorForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, stDesc, theta, grid); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSpatialTfGridGeneratorBackward( |
| cudnnHandle_t handle, |
| const cudnnSpatialTransformerDescriptor_t stDesc, |
| const void *dgrid, |
| void *dtheta) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnSpatialTransformerDescriptor_t, const void *, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSpatialTfGridGeneratorBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, stDesc, dgrid, dtheta); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSpatialTfSamplerForward( |
| cudnnHandle_t handle, |
| cudnnSpatialTransformerDescriptor_t stDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *grid, |
| const void *beta, |
| cudnnTensorDescriptor_t yDesc, |
| void *y) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnSpatialTransformerDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const void *, cudnnTensorDescriptor_t, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSpatialTfSamplerForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, stDesc, alpha, xDesc, x, grid, beta, yDesc, y); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSpatialTfSamplerBackward( |
| cudnnHandle_t handle, |
| cudnnSpatialTransformerDescriptor_t stDesc, |
| const void *alpha, |
| const cudnnTensorDescriptor_t xDesc, |
| const void *x, |
| const void *beta, |
| const cudnnTensorDescriptor_t dxDesc, |
| void *dx, |
| const void *alphaDgrid, |
| const cudnnTensorDescriptor_t dyDesc, |
| const void *dy, |
| const void *grid, |
| const void *betaDgrid, |
| void *dgrid) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnSpatialTransformerDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const cudnnTensorDescriptor_t, void *, const void *, const cudnnTensorDescriptor_t, const void *, const void *, const void *, void *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSpatialTfSamplerBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, stDesc, alpha, xDesc, x, beta, dxDesc, dx, alphaDgrid, dyDesc, dy, grid, betaDgrid, dgrid); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateDropoutDescriptor(cudnnDropoutDescriptor_t * dropoutDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnDropoutDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateDropoutDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(dropoutDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyDropoutDescriptor(cudnnDropoutDescriptor_t dropoutDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnDropoutDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyDropoutDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(dropoutDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDropoutGetStatesSize(cudnnHandle_t handle, size_t * sizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDropoutGetStatesSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDropoutGetReserveSpaceSize(cudnnTensorDescriptor_t xdesc, size_t * sizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnTensorDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDropoutGetReserveSpaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(xdesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetDropoutDescriptor(cudnnDropoutDescriptor_t dropoutDesc, |
| cudnnHandle_t handle, |
| float dropout, |
| void * states, |
| size_t stateSizeInBytes, |
| unsigned long long seed) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnDropoutDescriptor_t, cudnnHandle_t, float, void *, size_t, unsigned long long); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetDropoutDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(dropoutDesc, handle, dropout, states, stateSizeInBytes, seed); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRestoreDropoutDescriptor(cudnnDropoutDescriptor_t dropoutDesc, |
| cudnnHandle_t handle, |
| float dropout, |
| void * states, |
| size_t stateSizeInBytes, |
| unsigned long long seed) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnDropoutDescriptor_t, cudnnHandle_t, float, void *, size_t, unsigned long long); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRestoreDropoutDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(dropoutDesc, handle, dropout, states, stateSizeInBytes, seed); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetDropoutDescriptor(cudnnDropoutDescriptor_t dropoutDesc, |
| cudnnHandle_t handle, |
| float * dropout, |
| void ** states, |
| unsigned long long * seed) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnDropoutDescriptor_t, cudnnHandle_t, float *, void **, unsigned long long *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetDropoutDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(dropoutDesc, handle, dropout, states, seed); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDropoutForward(cudnnHandle_t handle, |
| const cudnnDropoutDescriptor_t dropoutDesc, |
| const cudnnTensorDescriptor_t xdesc, |
| const void * x, |
| const cudnnTensorDescriptor_t ydesc, |
| void * y, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnDropoutDescriptor_t, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, void *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDropoutForward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, dropoutDesc, xdesc, x, ydesc, y, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDropoutBackward(cudnnHandle_t handle, |
| const cudnnDropoutDescriptor_t dropoutDesc, |
| const cudnnTensorDescriptor_t dydesc, |
| const void * dy, |
| const cudnnTensorDescriptor_t dxdesc, |
| void * dx, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnDropoutDescriptor_t, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, void *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDropoutBackward"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, dropoutDesc, dydesc, dy, dxdesc, dx, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateRNNDescriptor(cudnnRNNDescriptor_t * rnnDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateRNNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyRNNDescriptor(cudnnRNNDescriptor_t rnnDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyRNNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNForwardInferenceAlgorithmMaxCount( |
| cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNForwardInferenceAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindRNNForwardInferenceAlgorithmEx( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t * xDesc, |
| const void * x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void * hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void * cx, |
| const cudnnFilterDescriptor_t wDesc, |
| const void * w, |
| const cudnnTensorDescriptor_t *yDesc, |
| void * y, |
| const cudnnTensorDescriptor_t hyDesc, |
| void * hy, |
| const cudnnTensorDescriptor_t cyDesc, |
| void * cy, |
| const float findIntensity, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnAlgorithmPerformance_t *perfResults, |
| void * workspace, |
| size_t workSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, const float, const int, int *, cudnnAlgorithmPerformance_t *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindRNNForwardInferenceAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, cxDesc, cx, wDesc, w, yDesc, y, hyDesc, hy, cyDesc, cy, findIntensity, requestedAlgoCount, returnedAlgoCount, perfResults, workspace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNForwardTrainingAlgorithmMaxCount( |
| cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNForwardTrainingAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindRNNForwardTrainingAlgorithmEx( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t * xDesc, |
| const void * x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void * hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void * cx, |
| const cudnnFilterDescriptor_t wDesc, |
| const void * w, |
| const cudnnTensorDescriptor_t *yDesc, |
| void * y, |
| const cudnnTensorDescriptor_t hyDesc, |
| void * hy, |
| const cudnnTensorDescriptor_t cyDesc, |
| void * cy, |
| const float findIntensity, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnAlgorithmPerformance_t *perfResults, |
| void * workspace, |
| size_t workSpaceSizeInBytes, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, const float, const int, int *, cudnnAlgorithmPerformance_t *, void *, size_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindRNNForwardTrainingAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, cxDesc, cx, wDesc, w, yDesc, y, hyDesc, hy, cyDesc, cy, findIntensity, requestedAlgoCount, returnedAlgoCount, perfResults, workspace, workSpaceSizeInBytes, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNBackwardDataAlgorithmMaxCount( |
| cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNBackwardDataAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindRNNBackwardDataAlgorithmEx( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t * yDesc, |
| const void * y, |
| const cudnnTensorDescriptor_t * dyDesc, |
| const void * dy, |
| const cudnnTensorDescriptor_t dhyDesc, |
| const void * dhy, |
| const cudnnTensorDescriptor_t dcyDesc, |
| const void * dcy, |
| const cudnnFilterDescriptor_t wDesc, |
| const void * w, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void * hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void * cx, |
| const cudnnTensorDescriptor_t * dxDesc, |
| void * dx, |
| const cudnnTensorDescriptor_t dhxDesc, |
| void * dhx, |
| const cudnnTensorDescriptor_t dcxDesc, |
| void * dcx, |
| const float findIntensity, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnAlgorithmPerformance_t *perfResults, |
| void * workspace, |
| size_t workSpaceSizeInBytes, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, const float, const int, int *, cudnnAlgorithmPerformance_t *, void *, size_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindRNNBackwardDataAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, yDesc, y, dyDesc, dy, dhyDesc, dhy, dcyDesc, dcy, wDesc, w, hxDesc, hx, cxDesc, cx, dxDesc, dx, dhxDesc, dhx, dcxDesc, dcx, findIntensity, requestedAlgoCount, returnedAlgoCount, perfResults, workspace, workSpaceSizeInBytes, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNBackwardWeightsAlgorithmMaxCount( |
| cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| int *count) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNBackwardWeightsAlgorithmMaxCount"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, count); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnFindRNNBackwardWeightsAlgorithmEx( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t * xDesc, |
| const void * x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void * hx, |
| const cudnnTensorDescriptor_t * yDesc, |
| const void * y, |
| const float findIntensity, |
| const int requestedAlgoCount, |
| int *returnedAlgoCount, |
| cudnnAlgorithmPerformance_t *perfResults, |
| const void * workspace, |
| size_t workSpaceSizeInBytes, |
| const cudnnFilterDescriptor_t dwDesc, |
| void * dw, |
| const void * reserveSpace, |
| size_t reserveSpaceSizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t *, const void *, const float, const int, int *, cudnnAlgorithmPerformance_t *, const void *, size_t, const cudnnFilterDescriptor_t, void *, const void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnFindRNNBackwardWeightsAlgorithmEx"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, yDesc, y, findIntensity, requestedAlgoCount, returnedAlgoCount, perfResults, workspace, workSpaceSizeInBytes, dwDesc, dw, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreatePersistentRNNPlan(cudnnRNNDescriptor_t rnnDesc, |
| const int minibatch, |
| const cudnnDataType_t dataType, |
| cudnnPersistentRNNPlan_t * plan) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t, const int, const cudnnDataType_t, cudnnPersistentRNNPlan_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreatePersistentRNNPlan"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc, minibatch, dataType, plan); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetPersistentRNNPlan(cudnnRNNDescriptor_t rnnDesc, |
| cudnnPersistentRNNPlan_t plan) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t, cudnnPersistentRNNPlan_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetPersistentRNNPlan"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc, plan); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyPersistentRNNPlan(cudnnPersistentRNNPlan_t plan) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnPersistentRNNPlan_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyPersistentRNNPlan"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(plan); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNDescriptor(cudnnHandle_t handle, |
| cudnnRNNDescriptor_t rnnDesc, |
| const int hiddenSize, |
| const int numLayers, |
| cudnnDropoutDescriptor_t dropoutDesc, /* Between layers, not between recurrent steps. */ |
| cudnnRNNInputMode_t inputMode, |
| cudnnDirectionMode_t direction, |
| cudnnRNNMode_t mode, |
| cudnnRNNAlgo_t algo, |
| cudnnDataType_t dataType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnRNNDescriptor_t, const int, const int, cudnnDropoutDescriptor_t, cudnnRNNInputMode_t, cudnnDirectionMode_t, cudnnRNNMode_t, cudnnRNNAlgo_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, algo, dataType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNProjectionLayers(cudnnHandle_t handle, |
| cudnnRNNDescriptor_t rnnDesc, |
| const int recProjSize, |
| const int outProjSize) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnRNNDescriptor_t, const int, const int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNProjectionLayers"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, recProjSize, outProjSize); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNProjectionLayers(cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| int *recProjSize, |
| int *outProjSize) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, int *, int *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNProjectionLayers"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, recProjSize, outProjSize); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNAlgorithmDescriptor(cudnnHandle_t handle, |
| cudnnRNNDescriptor_t rnnDesc, |
| cudnnAlgorithmDescriptor_t algoDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnRNNDescriptor_t, cudnnAlgorithmDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, algoDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNDescriptor(cudnnHandle_t handle, |
| cudnnRNNDescriptor_t rnnDesc, |
| int * hiddenSize, |
| int * numLayers, |
| cudnnDropoutDescriptor_t * dropoutDesc, |
| cudnnRNNInputMode_t * inputMode, |
| cudnnDirectionMode_t * direction, |
| cudnnRNNMode_t * mode, |
| cudnnRNNAlgo_t * algo, |
| cudnnDataType_t * dataType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnRNNDescriptor_t, int *, int *, cudnnDropoutDescriptor_t *, cudnnRNNInputMode_t *, cudnnDirectionMode_t *, cudnnRNNMode_t *, cudnnRNNAlgo_t *, cudnnDataType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, algo, dataType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNMatrixMathType(cudnnRNNDescriptor_t rnnDesc, cudnnMathType_t mType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t, cudnnMathType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNMatrixMathType"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc, mType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNMatrixMathType(cudnnRNNDescriptor_t rnnDesc, cudnnMathType_t* mType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t, cudnnMathType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNMatrixMathType"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc, mType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNWorkspaceSize( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *xDesc, |
| size_t *sizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNTrainingReserveSize( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *xDesc, |
| size_t *sizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNTrainingReserveSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNParamsSize(cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const cudnnTensorDescriptor_t xDesc, |
| size_t *sizeInBytes, |
| cudnnDataType_t dataType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const cudnnTensorDescriptor_t, size_t *, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNParamsSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, xDesc, sizeInBytes, dataType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNLinLayerMatrixParams( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int pseudoLayer, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const int linLayerID, |
| cudnnFilterDescriptor_t linLayerMatDesc, |
| void **linLayerMat) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const void *, const int, cudnnFilterDescriptor_t, void **); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNLinLayerMatrixParams"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, pseudoLayer, xDesc, wDesc, w, linLayerID, linLayerMatDesc, linLayerMat); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetRNNLinLayerBiasParams( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int pseudoLayer, |
| const cudnnTensorDescriptor_t xDesc, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const int linLayerID, |
| cudnnFilterDescriptor_t linLayerBiasDesc, |
| void **linLayerBias) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t, const cudnnFilterDescriptor_t, const void *, const int, cudnnFilterDescriptor_t, void **); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetRNNLinLayerBiasParams"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, pseudoLayer, xDesc, wDesc, w, linLayerID, linLayerBiasDesc, linLayerBias); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRNNForwardInference( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *xDesc, |
| const void *x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void *hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void *cx, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnTensorDescriptor_t *yDesc, |
| void *y, |
| const cudnnTensorDescriptor_t hyDesc, |
| void *hy, |
| const cudnnTensorDescriptor_t cyDesc, |
| void *cy, |
| void *workspace, |
| size_t workSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRNNForwardInference"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, cxDesc, cx, wDesc, w, yDesc, y, hyDesc, hy, cyDesc, cy, workspace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRNNForwardTraining( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *xDesc, |
| const void *x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void *hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void *cx, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnTensorDescriptor_t *yDesc, |
| void *y, |
| const cudnnTensorDescriptor_t hyDesc, |
| void *hy, |
| const cudnnTensorDescriptor_t cyDesc, |
| void *cy, |
| void *workspace, |
| size_t workSpaceSizeInBytes, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, void *, size_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRNNForwardTraining"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, cxDesc, cx, wDesc, w, yDesc, y, hyDesc, hy, cyDesc, cy, workspace, workSpaceSizeInBytes, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRNNBackwardData( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *yDesc, |
| const void *y, |
| const cudnnTensorDescriptor_t *dyDesc, |
| const void *dy, |
| const cudnnTensorDescriptor_t dhyDesc, |
| const void *dhy, |
| const cudnnTensorDescriptor_t dcyDesc, |
| const void *dcy, |
| const cudnnFilterDescriptor_t wDesc, |
| const void *w, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void *hx, |
| const cudnnTensorDescriptor_t cxDesc, |
| const void *cx, |
| const cudnnTensorDescriptor_t *dxDesc, |
| void *dx, |
| const cudnnTensorDescriptor_t dhxDesc, |
| void *dhx, |
| const cudnnTensorDescriptor_t dcxDesc, |
| void *dcx, |
| void *workspace, |
| size_t workSpaceSizeInBytes, |
| void * reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnFilterDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t *, void *, const cudnnTensorDescriptor_t, void *, const cudnnTensorDescriptor_t, void *, void *, size_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRNNBackwardData"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, yDesc, y, dyDesc, dy, dhyDesc, dhy, dcyDesc, dcy, wDesc, w, hxDesc, hx, cxDesc, cx, dxDesc, dx, dhxDesc, dhx, dcxDesc, dcx, workspace, workSpaceSizeInBytes, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRNNBackwardWeights( cudnnHandle_t handle, |
| const cudnnRNNDescriptor_t rnnDesc, |
| const int seqLength, |
| const cudnnTensorDescriptor_t *xDesc, |
| const void *x, |
| const cudnnTensorDescriptor_t hxDesc, |
| const void *hx, |
| const cudnnTensorDescriptor_t *yDesc, |
| const void *y, |
| const void *workspace, |
| size_t workSpaceSizeInBytes, |
| const cudnnFilterDescriptor_t dwDesc, |
| void *dw, |
| const void *reserveSpace, |
| size_t reserveSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnRNNDescriptor_t, const int, const cudnnTensorDescriptor_t *, const void *, const cudnnTensorDescriptor_t, const void *, const cudnnTensorDescriptor_t *, const void *, const void *, size_t, const cudnnFilterDescriptor_t, void *, const void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRNNBackwardWeights"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, seqLength, xDesc, x, hxDesc, hx, yDesc, y, workspace, workSpaceSizeInBytes, dwDesc, dw, reserveSpace, reserveSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateCTCLossDescriptor( cudnnCTCLossDescriptor_t* ctcLossDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnCTCLossDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateCTCLossDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(ctcLossDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetCTCLossDescriptor( |
| cudnnCTCLossDescriptor_t ctcLossDesc, |
| cudnnDataType_t compType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnCTCLossDescriptor_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetCTCLossDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(ctcLossDesc, compType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetCTCLossDescriptor( |
| cudnnCTCLossDescriptor_t ctcLossDesc, |
| cudnnDataType_t* compType ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnCTCLossDescriptor_t, cudnnDataType_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetCTCLossDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(ctcLossDesc, compType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyCTCLossDescriptor( cudnnCTCLossDescriptor_t ctcLossDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnCTCLossDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyCTCLossDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(ctcLossDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCTCLoss( cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t probsDesc, /* Tensor descriptor for probabilities, the dimensions are T,N,A (T is the timing steps, N is the mini batch size, A is the alphabet size) */ |
| const void * probs, /* probabilities after softmax, in GPU memory */ |
| const int * labels, /* labels, in CPU memory */ |
| const int * labelLengths, /* the length of each label, in CPU memory */ |
| const int * inputLengths, /* the lengths of timing steps in each batch, in CPU memory */ |
| void * costs, /* the returned costs of CTC, in GPU memory */ |
| const cudnnTensorDescriptor_t gradientsDesc, /* Tensor descriptor for gradients, the dimensions are T,N,A */ |
| const void * gradients, /* the returned CTC gradients, in GPU memory, to compute costs only, set it to NULL */ |
| cudnnCTCLossAlgo_t algo, /* algorithm selected, supported now 0 and 1 */ |
| cudnnCTCLossDescriptor_t ctcLossDesc, |
| void * workspace, /* pointer to the workspace, in GPU memory */ |
| size_t workSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const void *, const int *, const int *, const int *, void *, const cudnnTensorDescriptor_t, const void *, cudnnCTCLossAlgo_t, cudnnCTCLossDescriptor_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCTCLoss"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, probsDesc, probs, labels, labelLengths, inputLengths, costs, gradientsDesc, gradients, algo, ctcLossDesc, workspace, workSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetCTCLossWorkspaceSize( |
| cudnnHandle_t handle, |
| const cudnnTensorDescriptor_t probsDesc, /* Tensor descriptor for probabilities, the dimensions are T,N,A (T is the timing steps, N is the mini batch size, A is the alphabet size) */ |
| const cudnnTensorDescriptor_t gradientsDesc, /* Tensor descriptor for gradients, the dimensions are T,N,A. To compute costs only, set it to NULL */ |
| const int * labels, /* labels, in CPU memory */ |
| const int * labelLengths, /* the length of each label, in CPU memory */ |
| const int * inputLengths, /* the lengths of timing steps in each batch, in CPU memory */ |
| cudnnCTCLossAlgo_t algo, /* algorithm selected, supported now 0 and 1 */ |
| cudnnCTCLossDescriptor_t ctcLossDesc, |
| size_t *sizeInBytes ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, const cudnnTensorDescriptor_t, const cudnnTensorDescriptor_t, const int *, const int *, const int *, cudnnCTCLossAlgo_t, cudnnCTCLossDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetCTCLossWorkspaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, probsDesc, gradientsDesc, labels, labelLengths, inputLengths, algo, ctcLossDesc, sizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateAlgorithmDescriptor( |
| cudnnAlgorithmDescriptor_t *algoDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmDescriptor_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetAlgorithmDescriptor( |
| cudnnAlgorithmDescriptor_t algoDesc, |
| cudnnAlgorithm_t algorithm) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmDescriptor_t, cudnnAlgorithm_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoDesc, algorithm); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetAlgorithmDescriptor( |
| const cudnnAlgorithmDescriptor_t algoDesc, |
| cudnnAlgorithm_t* algorithm) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnAlgorithmDescriptor_t, cudnnAlgorithm_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoDesc, algorithm); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCopyAlgorithmDescriptor( |
| const cudnnAlgorithmDescriptor_t src, |
| cudnnAlgorithmDescriptor_t dest) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnAlgorithmDescriptor_t, cudnnAlgorithmDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCopyAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(src, dest); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyAlgorithmDescriptor( |
| cudnnAlgorithmDescriptor_t algoDesc ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyAlgorithmDescriptor"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnCreateAlgorithmPerformance( |
| cudnnAlgorithmPerformance_t* algoPerf, |
| int numberToCreate ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmPerformance_t *, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnCreateAlgorithmPerformance"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoPerf, numberToCreate); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetAlgorithmPerformance( |
| cudnnAlgorithmPerformance_t algoPerf, |
| cudnnAlgorithmDescriptor_t algoDesc, |
| cudnnStatus_t status, |
| float time, |
| size_t memory ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmPerformance_t, cudnnAlgorithmDescriptor_t, cudnnStatus_t, float, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetAlgorithmPerformance"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoPerf, algoDesc, status, time, memory); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetAlgorithmPerformance( |
| const cudnnAlgorithmPerformance_t algoPerf, |
| cudnnAlgorithmDescriptor_t* algoDesc, |
| cudnnStatus_t* status, |
| float* time, |
| size_t* memory ) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnAlgorithmPerformance_t, cudnnAlgorithmDescriptor_t *, cudnnStatus_t *, float *, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetAlgorithmPerformance"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoPerf, algoDesc, status, time, memory); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnDestroyAlgorithmPerformance( |
| cudnnAlgorithmPerformance_t* algoPerf, |
| int numberToDestroy) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnAlgorithmPerformance_t *, int); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnDestroyAlgorithmPerformance"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(algoPerf, numberToDestroy); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetAlgorithmSpaceSize( |
| cudnnHandle_t handle, |
| cudnnAlgorithmDescriptor_t algoDesc, |
| size_t* algoSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnAlgorithmDescriptor_t, size_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetAlgorithmSpaceSize"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, algoDesc, algoSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSaveAlgorithm( |
| cudnnHandle_t handle, |
| cudnnAlgorithmDescriptor_t algoDesc, |
| void* algoSpace, |
| size_t algoSpaceSizeInBytes) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnAlgorithmDescriptor_t, void *, size_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSaveAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, algoDesc, algoSpace, algoSpaceSizeInBytes); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnRestoreAlgorithm( |
| cudnnHandle_t handle, |
| void* algoSpace, |
| size_t algoSpaceSizeInBytes, |
| cudnnAlgorithmDescriptor_t algoDesc) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, void *, size_t, cudnnAlgorithmDescriptor_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnRestoreAlgorithm"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, algoSpace, algoSpaceSizeInBytes, algoDesc); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetCallback( |
| unsigned mask, |
| void *udata, |
| cudnnCallback_t fptr) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(unsigned int, void *, cudnnCallback_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetCallback"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(mask, udata, fptr); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnGetCallback( |
| unsigned *mask, |
| void **udata, |
| cudnnCallback_t *fptr) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(unsigned int *, void **, cudnnCallback_t *); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetCallback"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(mask, udata, fptr); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNDescriptor_v6(cudnnHandle_t handle, |
| cudnnRNNDescriptor_t rnnDesc, |
| const int hiddenSize, |
| const int numLayers, |
| cudnnDropoutDescriptor_t dropoutDesc, |
| cudnnRNNInputMode_t inputMode, |
| cudnnDirectionMode_t direction, |
| cudnnRNNMode_t mode, |
| cudnnRNNAlgo_t algo, |
| cudnnDataType_t dataType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnHandle_t, cudnnRNNDescriptor_t, const int, const int, cudnnDropoutDescriptor_t, cudnnRNNInputMode_t, cudnnDirectionMode_t, cudnnRNNMode_t, cudnnRNNAlgo_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNDescriptor_v6"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(handle, rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, algo, dataType); |
| } |
| |
| cudnnStatus_t CUDNNWINAPI cudnnSetRNNDescriptor_v5(cudnnRNNDescriptor_t rnnDesc, |
| int hiddenSize, |
| int numLayers, |
| cudnnDropoutDescriptor_t dropoutDesc, |
| cudnnRNNInputMode_t inputMode, |
| cudnnDirectionMode_t direction, |
| cudnnRNNMode_t mode, |
| cudnnDataType_t dataType) { |
| using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnRNNDescriptor_t, int, int, cudnnDropoutDescriptor_t, cudnnRNNInputMode_t, cudnnDirectionMode_t, cudnnRNNMode_t, cudnnDataType_t); |
| static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetRNNDescriptor_v5"); |
| if (!func_ptr) return GetSymbolNotFoundError(); |
| return func_ptr(rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, dataType); |
| } |
| |
| } // extern "C" |