Skip to main content

launch_depthwise

cubek_convolution::kernels::forward::depthwise

Function launch_depthwise 

Source 
pub fn launch_depthwise(
    client: &Client,
    tensors: DepthwiseTensors,
    args: ConvolutionArgs<2>,
    groups: usize,
    dtype: ElemType,
    strategy: DepthwiseStrategy,
) -> Result<(), ConvSetupError>
Expand description

Launch a depthwise convolution under strategy.

§Errors

ConvSetupError::NotDepthwise when the convolution is not one filter per channel — the tuner reads that as “this candidate does not apply here”, which is exactly what it is — and ConvSetupError::InvalidConfig when a fixed tiling is degenerate. Returns ConvSetupError::Unknown when re-laying the filter cannot be launched.