Skip to main content

launch_kernel_concrete

cubek_convolution::kernels::backward_data::selector

Function launch_kernel_concrete 

Source 
pub fn launch_kernel_concrete<R: Runtime, Args: ConcreteArgs<A>, A: Routine<RuntimeArgs>>(
    client: &ComputeClient<R>,
    out_grad: InputBinding<R>,
    weights: InputBinding<R>,
    in_grad: TensorBinding<R>,
    problem: ConvolutionProblem,
    vector_sizes: MatmulVectorSizes,
    blueprint_strategy: &BlueprintStrategy<RuntimeArgs, A>,
    dtypes: &MatmulElems,
) -> Result<(), ConvSetupError>
Expand description

Select which kernel to launch for the given Algorithm.

Only works for concrete tensor inputs and output.