cubecl_linalg::convolution::base

Trait ConvolutionLaunch

Source 
pub trait ConvolutionLaunch: ConvolutionConfigFactory {
    // 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>,
        bias: Option<TensorArg<'a, R>>,
        output: OutputRuntimeArg<'a, MS, R>,
        problem: &ConvolutionProblem,
        config: <Self as ConvolutionConfigFactory>::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>, bias: Option<TensorArg<'a, R>>, output: OutputRuntimeArg<'a, MS, R>, problem: &ConvolutionProblem, config: <Self as ConvolutionConfigFactory>::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§