Skip to main content

launch_kernel_concrete

cubek_convolution::kernels::forward::selector

Function launch_kernel_concrete 

Source 
pub fn launch_kernel_concrete<Args: ConcreteArgs<A>, A: BatchMatmulRoutine<RuntimeArgs>>(
    client: &Client,
    input: InputBinding,
    weight: InputBinding,
    bias: Option<InputBinding>,
    out: TensorBinding,
    problem: ConvolutionProblem,
    vector_sizes: MatmulVectorSizes,
    blueprint_strategy: &BlueprintStrategy<Args::Config, A>,
    dtypes: &MatmulElems,
) -> Result<(), ConvSetupError>
Expand description

Select which kernel to launch for the given Algorithm.

Only works for concrete tensor inputs and output.