Struct OneToManyMatmulFamily 

pub struct OneToManyMatmulFamily<GMM: GlobalMatmulFamily, S: SpanMatmul, C: CubeDispatch> { /* private fields */ }

Trait Implementations

impl<GMM: GlobalMatmulFamily, S: SpanMatmul, C: CubeDispatch> BatchMatmulFamily for OneToManyMatmulFamily<GMM, S, C>

type Matmul<MP: MatmulPrecision> = OneToManyMatmul<MP, <GMM as GlobalMatmulFamily>::Matmul<MP>, S, C>

impl<GMM: GlobalMatmulFamily, S: SpanMatmul, C: CubeDispatch> MatmulConfigFactory for OneToManyMatmulFamily<GMM, S, C>

type Config = Config<<GMM as MatmulConfigFactory>::Config, C>

Configuration tailored to the matmul implementation

type Input = <GMM as MatmulConfigFactory>::Input

fn check_config(config: &Self::Config) -> Result<(), InvalidConfigError>

Asserts that the configuration for this matmul will lead to a valid computation

fn check_availability<R: Runtime, MP: MatmulPrecision>( client: &ComputeClient<R::Server, R::Channel>, config: &Self::Config, ) -> Result<(), MatmulAvailabilityError>

Checks if the client can handle the features used in this computation

fn make_config( input: Self::Input, problem: &MatmulProblem, cube_dim: &CubeDim, cube_count: &CubeCount, quantized: bool, ) -> Self::Config

Create config for this matmul, given launch information

impl<GMM: GlobalMatmulFamily, S: SpanMatmul, C: CubeDispatch> MatmulLaunch for OneToManyMatmulFamily<GMM, S, C>

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::Config, )

Entry point