Skip to main content

launch_kernel

cubek_convolution::kernels::backward_data::launch

Function launch_kernel 

Source 
pub fn launch_kernel<Rt: Routine>(
    client: &Client,
    out_grad: InputBinding,
    weights: InputBinding,
    in_grad: TensorBinding,
    problem: ConvolutionProblem,
    blueprint_strategy: &BlueprintStrategy<RuntimeArgs, Rt::MatmulRoutine>,
    dtypes: MatmulElems,
) -> Result<(), ConvSetupError>
where
    Rt::Args: ConcreteArgs<Rt::MatmulRoutine>,