cubecl_matmul::components::stage

Struct UnitMatmulFamily

Source 
pub struct UnitMatmulFamily<TMM: TileMatmulFamily, RF: ReaderFamily> { /* private fields */ }
Expand description

Unit Matmul family for any precision

Trait Implementations§

Source§

impl<TMM: TileMatmulFamily, RF: ReaderFamily> StageMatmulFamily for UnitMatmulFamily<TMM, RF>

Source§

type LhsReader = RF

Reader family for Lhs
Source§

type RhsReader = RF

Reader family for Rhs
Source§

type Matmul<MP: MatmulPrecision, TL: TilingLayout, TR: TilingLayout> = PartitionedStageMatmul<MP, <TMM as TileMatmulFamily>::Matmul<MP>, <RF as ReaderFamily>::Reader<<MP as MatmulPrecision>::ES, TL>, <RF as ReaderFamily>::Reader<<MP as MatmulPrecision>::ES, TR>, UnitPartitioner, UnitPartitionedStageConfig<<<TMM as TileMatmulFamily>::Matmul<MP> as TileMatmul<MP>>::Config>>

The specific [TileMatmul] implementation associated with this family.
Source§

type Config = UnitPartitionedStageConfig<<TMM as TileMatmulFamily>::Config>

The configuration type associated with this matmul family.
Source§

fn setup<MP: MatmulPrecision, R: Runtime>( client: &ComputeClient<R::Server, R::Channel>, problem: &MatmulProblem, selection: &MatmulSelection, line_sizes: &MatmulLineSizes, num_stages: NumStages, max_loaders: Option<MaxLoaderPlanes>, ordered: bool, ) -> Result<Self::Config, MatmulSetupError>

Constructs the configuration based on the matmul problem, selection, line sizes, number of stages, maximum of tasks per plane, and whether the algorithm is an ordered variant Read more
Source§

fn filter_line_sizes( available_line_sizes: AvailableLineSizes, ) -> AvailableLineSizes

Filters out line sizes that are incompatible with this matmul family. Read more

Auto Trait Implementations§

§

impl<TMM, RF> Freeze for UnitMatmulFamily<TMM, RF>

§

impl<TMM, RF> RefUnwindSafe for UnitMatmulFamily<TMM, RF>
where TMM: RefUnwindSafe, RF: RefUnwindSafe,

§

impl<TMM, RF> Send for UnitMatmulFamily<TMM, RF>

§

impl<TMM, RF> Sync for UnitMatmulFamily<TMM, RF>

§

impl<TMM, RF> Unpin for UnitMatmulFamily<TMM, RF>
where TMM: Unpin, RF: Unpin,

§

impl<TMM, RF> UnwindSafe for UnitMatmulFamily<TMM, RF>
where TMM: UnwindSafe, RF: UnwindSafe,

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V