What does torch.backends.cudnn.benchmark do?

Shubhankar January 13, 2020, 7:30pm 27

How to get (print) convolution_algorithm chosen by CUDNN autotune?

How to get the convolution algorithm chosen by CUDNN autotune in pytorch and how to manually define it later? Is it possible to interact with CUDNN API from inside Pytorch. Following CUDNN function returns the type of algorithm to be used defined by CUDNN_CONVOLUTION_FWD_PREFER_FASTEST: cudnnGetConvolutionForwardAlgorithm(cudnn, input_descriptor, kernel_descriptor, convoluti…

show post in topic

Home
Categories
Guidelines
Terms of Service
Privacy Policy

Powered by Discourse, best viewed with JavaScript enabled