/// <summary>
/// </summary>
public ConvolutionDescriptor()
{
_desc = new cudnnConvolutionDescriptor();
res = CudaDNNNativeMethods.cudnnCreateConvolutionDescriptor(ref _desc);
Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnCreateConvolutionDescriptor", res));
if (res != cudnnStatus.Success)
throw new CudaDNNException(res);
}
/// <summary>
/// </summary>
public ConvolutionDescriptor()
{
_desc = new cudnnConvolutionDescriptor();
res = CudaDNNNativeMethods.cudnnCreateConvolutionDescriptor(ref _desc);
Debug.WriteLine(String.Format("{0:G}, {1}: {2}", DateTime.Now, "cudnnCreateConvolutionDescriptor", res));
if (res != cudnnStatus.Success)
{
throw new CudaDNNException(res);
}
}
public static extern cudnnStatus cudnnGetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLengthRequested,
ref int arrayLength,
int[] padA,
int[] strideA,
int[] upscaleA,
ref cudnnConvolutionMode mode,
ref cudnnDataType dataType // convolution data type
);
public static extern cudnnStatus cudnnSetConvolutionNdDescriptor(cudnnConvolutionDescriptor convDesc,
int arrayLength, /* nbDims-2 size */
int[] padA,
int[] filterStrideA,
int[] upscaleA,
cudnnConvolutionMode mode,
cudnnDataType dataType // convolution data type
);
public static extern cudnnStatus cudnnGetConvolution2dForwardOutputDim( cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor inputTensorDesc, cudnnFilterDescriptor filterDesc, ref int n, ref int c, ref int h, ref int w );
public static extern cudnnStatus cudnnFindConvolutionForwardAlgorithmEx(
cudnnHandle handle,
cudnnTensorDescriptor xDesc,
CUdeviceptr x,
cudnnFilterDescriptor wDesc,
CUdeviceptr w,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor yDesc,
CUdeviceptr y,
int requestedAlgoCount,
ref int returnedAlgoCount,
ref cudnnConvolutionFwdAlgoPerf perfResults,
CUdeviceptr workSpace,
SizeT workSpaceSizeInBytes );
public static extern cudnnStatus cudnnGetConvolutionForwardWorkspaceSize( cudnnHandle handle, cudnnTensorDescriptor srcDesc, cudnnFilterDescriptor filterDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor destDesc, cudnnConvolutionFwdAlgo algo, ref SizeT sizeInBytes );
public static extern cudnnStatus cudnnFindConvolutionForwardAlgorithm(cudnnHandle handle,
cudnnTensorDescriptor srcDesc,
cudnnFilterDescriptor filterDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor destDesc,
int requestedAlgoCount,
ref int returnedAlgoCount,
cudnnConvolutionFwdAlgoPerf[] perfResults
);
public static extern cudnnStatus cudnnConvolutionBackwardData( cudnnHandle handle, ref double alpha, cudnnFilterDescriptor filterDesc, CUdeviceptr filterData, cudnnTensorDescriptor diffDesc, CUdeviceptr diffData, cudnnConvolutionDescriptor convDesc, cudnnConvolutionBwdDataAlgo algo, CUdeviceptr workSpace, SizeT workSpaceSizeInBytes, ref double beta, cudnnTensorDescriptor gradDesc, CUdeviceptr gradData );
public static extern cudnnStatus cudnnGetConvolutionBackwardDataWorkspaceSize( cudnnHandle handle, cudnnFilterDescriptor filterDesc, cudnnTensorDescriptor diffDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor gradDesc, cudnnConvolutionBwdDataAlgo algo, ref SizeT sizeInBytes );
public static extern cudnnStatus cudnnGetConvolutionBackwardDataAlgorithm( cudnnHandle handle, cudnnFilterDescriptor filterDesc, cudnnTensorDescriptor diffDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor gradDesc, cudnnConvolutionBwdDataPreference preference, SizeT memoryLimitInbytes, ref cudnnConvolutionBwdDataAlgo algo );
public static extern cudnnStatus cudnnFindConvolutionBackwardDataAlgorithm( cudnnHandle handle,
cudnnFilterDescriptor filterDesc,
cudnnTensorDescriptor diffDesc,
cudnnConvolutionDescriptor convDesc,
cudnnTensorDescriptor gradDesc,
int requestedAlgoCount,
ref int returnedAlgoCount,
cudnnConvolutionBwdDataAlgoPerf[] perfResults
);
public static extern cudnnStatus cudnnConvolutionBackwardFilter( cudnnHandle handle, ref float alpha, cudnnTensorDescriptor srcDesc, CUdeviceptr srcData, cudnnTensorDescriptor diffDesc, CUdeviceptr diffData, cudnnConvolutionDescriptor convDesc, ref float beta, cudnnFilterDescriptor gradDesc, CUdeviceptr gradData );
public static extern cudnnStatus cudnnSetConvolution2dDescriptor( cudnnConvolutionDescriptor convDesc,
int pad_h, // zero-padding height
int pad_w, // zero-padding width
int u, // vertical filter stride
int v, // horizontal filter stride
int upscalex, // upscale the input in x-direction
int upscaley, // upscale the input in y-direction
cudnnConvolutionMode mode,
cudnnDataType dataType
);
public static extern cudnnStatus cudnnGetConvolutionNdForwardOutputDim( cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor inputTensorDesc, cudnnFilterDescriptor filterDesc, int nbDims, int[] tensorOuputDimA );
public static extern cudnnStatus cudnnDestroyConvolutionDescriptor( cudnnConvolutionDescriptor convDesc );
public static extern cudnnStatus cudnnConvolutionBackwardData( cudnnHandle handle, ref double alpha, cudnnFilterDescriptor filterDesc, CUdeviceptr filterData, cudnnTensorDescriptor diffDesc, CUdeviceptr diffData, cudnnConvolutionDescriptor convDesc, ref double beta, cudnnTensorDescriptor gradDesc, CUdeviceptr gradData );
public static extern cudnnStatus cudnnGetConvolutionForwardAlgorithm( cudnnHandle handle, cudnnTensorDescriptor srcDesc, cudnnFilterDescriptor filterDesc, cudnnConvolutionDescriptor convDesc, cudnnTensorDescriptor destDesc, cudnnConvolutionFwdPreference preference, SizeT memoryLimitInbytes, ref cudnnConvolutionFwdAlgo algo );
public static extern cudnnStatus cudnnIm2Col( cudnnHandle handle, cudnnTensorDescriptor srcDesc, CUdeviceptr srcData, cudnnFilterDescriptor filterDesc, cudnnConvolutionDescriptor convDesc, CUdeviceptr colBuffer );
public static extern cudnnStatus cudnnConvolutionForward( cudnnHandle handle, ref double alpha, cudnnTensorDescriptor srcDesc, CUdeviceptr srcData, cudnnFilterDescriptor filterDesc, CUdeviceptr filterData, cudnnConvolutionDescriptor convDesc, cudnnConvolutionFwdAlgo algo, CUdeviceptr workSpace, SizeT workSpaceSizeInBytes, ref double beta, cudnnTensorDescriptor destDesc, CUdeviceptr destData );
public static extern cudnnStatus cudnnCreateConvolutionDescriptor(ref cudnnConvolutionDescriptor convDesc );
public static extern cudnnStatus cudnnGetConvolution2dDescriptor( cudnnConvolutionDescriptor convDesc, ref int pad_h, // zero-padding height ref int pad_w, // zero-padding width ref int u, // vertical filter stride ref int v, // horizontal filter stride ref int upscalex, // upscale the input in x-direction ref int upscaley, // upscale the input in y-direction ref cudnnConvolutionMode mode );
public static extern cudnnStatus cudnnConvolutionBackwardFilter_v3( cudnnHandle handle, ref float alpha, cudnnTensorDescriptor srcDesc, CUdeviceptr srcData, cudnnTensorDescriptor diffDesc, CUdeviceptr diffData, cudnnConvolutionDescriptor convDesc, cudnnConvolutionBwdFilterAlgo algo, CUdeviceptr workSpace, SizeT workSpaceSizeInBytes, ref float beta, cudnnFilterDescriptor gradDesc, CUdeviceptr gradData );
