I’m working on an application in which 3D convolution is the bottleneck, and would like to get the best performance possible with preferably an existing library that supports 3D convolution.
I tried using cuDNN and got it to work, however, the user guide stated that FFT-based option (CUDNN_CONVOLUTION_FWD_ALGO_FFT) isn’t supported for 3D convolution. And since FFT is known as one of the fastest ways to compute convolution I was wondering if anyone has answers to any of the following questions:
- Is there an existing library with out-of-the-box 3D FFT-based convolution?
- Will there be a support for 3D FFT-based convolution in future releases of cuDNN?
Thanks a lot!