MatmulLaunch

cubecl_linalg::matmul::components

Trait MatmulLaunch 

Source 
pub trait MatmulLaunch: MatmulConfigFactory {
    // Required method
    unsafe fn launch_unchecked<'a, MS: MatmulSpec, R: Runtime>(
        client: &ComputeClient<<R as Runtime>::Server, <R as Runtime>::Channel>,
        cube_dim: CubeDim,
        cube_count: CubeCount,
        input: InputRuntimeArg<'a, MS, R>,
        output: OutputRuntimeArg<'a, MS, R>,
        size_k: ScalarArg<u32>,
        config: <Self as MatmulConfigFactory>::Config,
    );
}
Expand description

Provides launch entry point to solve a matmul

Required Methods§

Source

unsafe fn launch_unchecked<'a, MS: MatmulSpec, R: Runtime>( client: &ComputeClient<<R as Runtime>::Server, <R as Runtime>::Channel>, cube_dim: CubeDim, cube_count: CubeCount, input: InputRuntimeArg<'a, MS, R>, output: OutputRuntimeArg<'a, MS, R>, size_k: ScalarArg<u32>, config: <Self as MatmulConfigFactory>::Config, )

Entry point

§Safety

Out-of-bounds can happen

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§