// 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 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 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 tensorOuputDimA[] ) {
|
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, tensorOuputDimA);
|
}
|
|
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 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 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 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 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 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 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 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 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_v6(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_v6");
|
if (!func_ptr) return GetSymbolNotFoundError();
|
return func_ptr(handle, rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, algo, dataType);
|
}
|
|
cudnnStatus_t CUDNNWINAPI cudnnSetRNNDescriptor(cudnnRNNDescriptor_t rnnDesc,
|
int hiddenSize,
|
int numLayers,
|
cudnnDropoutDescriptor_t dropoutDesc, // Between layers, not between recurrent steps.
|
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");
|
if (!func_ptr) return GetSymbolNotFoundError();
|
return func_ptr(rnnDesc, hiddenSize, numLayers, dropoutDesc, inputMode, direction, mode, dataType);
|
}
|
|
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 layer,
|
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, layer, xDesc, wDesc, w, linLayerID, linLayerMatDesc, linLayerMat);
|
}
|
|
cudnnStatus_t CUDNNWINAPI cudnnGetRNNLinLayerBiasParams( cudnnHandle_t handle,
|
const cudnnRNNDescriptor_t rnnDesc,
|
const int layer,
|
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, layer, 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 cudnnSetConvolution2dDescriptor_v4(
|
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 ) {
|
using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(cudnnConvolutionDescriptor_t, int, int, int, int, int, int, cudnnConvolutionMode_t);
|
static auto func_ptr = LoadSymbol<FuncPtr>("cudnnSetConvolution2dDescriptor_v4");
|
if (!func_ptr) return GetSymbolNotFoundError();
|
return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
|
}
|
|
cudnnStatus_t CUDNNWINAPI cudnnSetConvolution2dDescriptor_v5( 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_v5");
|
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_v4(
|
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 ) {
|
using FuncPtr = cudnnStatus_t (CUDNNWINAPI *)(const cudnnConvolutionDescriptor_t, int *, int *, int *, int *, int *, int *, cudnnConvolutionMode_t *);
|
static auto func_ptr = LoadSymbol<FuncPtr>("cudnnGetConvolution2dDescriptor_v4");
|
if (!func_ptr) return GetSymbolNotFoundError();
|
return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
|
}
|
|
cudnnStatus_t CUDNNWINAPI cudnnGetConvolution2dDescriptor_v5( 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_v5");
|
if (!func_ptr) return GetSymbolNotFoundError();
|
return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
|
}
|
|
} // extern "C"
|
