Skip to main content

launch_kernel_concrete

cubek_convolution::kernels::backward_data::selector

Function launch_kernel_concrete 

Source 
pub fn launch_kernel_concrete<Args: ConcreteArgs<A>, A: BatchMatmulRoutine<RuntimeArgs>>(
    client: &Client,
    out_grad: InputBinding,
    weights: InputBinding,
    in_grad: TensorBinding,
    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.