I’ve been working on developing a new convolution algorithm. During my benchmarks with cuDNN in PyTorch, I encountered an issue: I’m unable to determine which convolution algorithms were selected by cuDNN.
I reviewed the Conv_v8.cpp file but couldn’t find any relevant clues. Even after extracting the logs from cuDNN, I couldn’t locate any related information within the descriptors. The conv descriptor is as below:
i! convDesc: type=cudnnConvolutionDescriptor_t:
i! mode: type=cudnnConvolutionMode_t; val=CUDNN_CROSS_CORRELATION (1);
i! dataType: type=cudnnDataType_t; val=CUDNN_DATA_FLOAT (0);
i! mathType: type=cudnnMathType_t; val=CUDNN_DEFAULT_MATH (0);
i! reorderType: type=int; val=0;
i! arrayLength: type=int; val=2;
i! padA: type=int; val=[1,1];
i! strideA: type=int; val=[1,1];
i! dilationA: type=int; val=[1,1];
i! groupCount: type=int; val=1;
Would anyone be able to assist me in determining how to identify the convolution algorithm selected by cuDNN within PyTorch?