ConvolutionLaunch

cubecl_convolution::components::global::entry_point

Trait ConvolutionLaunch 

Source 
pub trait ConvolutionLaunch<Config> {
    // Required method
    unsafe fn launch_unchecked<'a, MA: MatmulArgs, R: Runtime>(
        client: &ComputeClient<R>,
        cube_dim: CubeDim,
        cube_count: CubeCount,
        input: InputRuntimeArg<'a, MA, R>,
        output: OutputRuntimeArg<'a, MA, R>,
        problem: &ConvolutionProblem,
        config: Config,
        dtypes: &MatmulElems,
    ) -> Result<(), LaunchError>;
}
Expand description

Provides launch entry point to solve a matmul

Required Methods§

Source

unsafe fn launch_unchecked<'a, MA: MatmulArgs, R: Runtime>( client: &ComputeClient<R>, cube_dim: CubeDim, cube_count: CubeCount, input: InputRuntimeArg<'a, MA, R>, output: OutputRuntimeArg<'a, MA, R>, problem: &ConvolutionProblem, config: Config, dtypes: &MatmulElems, ) -> Result<(), LaunchError>

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§