Skip to main content

launch_ref

Function launch_ref 

Source
pub fn launch_ref<const N_SPATIAL: usize>(
    strategy: &Strategy,
    client: &Client,
    inputs: ConvolutionInputs,
    args: ConvolutionArgs<N_SPATIAL>,
    dtypes: MatmulElems,
) -> Result<(), ConvSetupError>
Expand description

The single public convolution entry point.

Routes the inputs (whose discriminant is the operation) and strategy (algorithm + tile-matmul kind, optionally a forced blueprint) into the right generic Routine and per-operation launch helper.