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

[xychan1998]

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”?

[AakankshaS]

Hi @xychan1998 ,

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

@xychan1998

If so, why not consider the extra workspace of “bias”/“activation”, or the input param “z” & “bias”?

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

Thanks