cubecl_reduce::reduce_kernel

Function launch_unchecked

Source 
pub unsafe fn launch_unchecked<'kernel, In: Numeric, Out: Numeric, Acc: Numeric, R: ReduceFamily, RA: ReduceArgs, __R: Runtime>(
    __client: &ComputeClient<__R::Server, __R::Channel>,
    __cube_count: CubeCount,
    __cube_dim: CubeDim,
    input: RuntimeArg<'kernel, RA::Input<In>, __R>,
    output: RuntimeArg<'kernel, RA::Output<Out>, __R>,
    axis_reduce: RuntimeArg<'kernel, u32, __R>,
    params: ReduceParams,
    config: R::Config,
)
Expand description

Launch the kernel reduce_kernel() on the given runtime