cudnn-frontend icon indicating copy to clipboard operation
cudnn-frontend copied to clipboard

[Question] How is the convolution algorithm set, and how to specify group number ?

Open haoqian-hao opened this issue 6 months ago • 0 comments

Hello all,

I'm currently working with convolutional layers in cudnn - python, and I have a couple of questions regarding the convolution algorithm selection and the setting of group numbers.

Convolution Algorithm Selection:

I would like to know how the convolution algorithm is chosen in the backend? Is there a way to explicitly set the algorithm, or is it automatically selected e.g., select runtime performance after evaluating all available algorithms?

Setting Group Numbers:

I'm working with grouped convolutions, and I'm trying to better understand how to set the group number correctly. I am wondering if there are some guidance for setting the group number?

Thank you in advance for your assistance!

haoqian-hao avatar Aug 26 '24 02:08 haoqian-hao