How to get workspace size of "cudnnConvolutionBiasActivationForward()"?

How to get workspace size of “cudnnConvolutionBiasActivationForward”?
Use “cudnnGetConvolutionForwardWorkspaceSize”?
If so, why not consider the extra workspace of “bias”/“activation”, or the input param “z” & “bias”?

Hi @xychan1998 ,

For legacy reasons, it is cudnnGetConvolutionForwardWorkspaceSize, there’s not a specialized get workspace size API for convBiasAct.

it is already considered inside cudnnGetConvolutionForwardWorkspaceSize. We suggest to use the backend API for more accurate workspace calculation.