Struct PlaneMatmulFamily

Source
pub struct PlaneMatmulFamily<TMM: TileMatmulFamily, LRF: ReaderFamily, RRF: ReaderFamily> { /* private fields */ }
Expand description

Plane Matmul family for any precision

Trait Implementations§

Source§

impl<TMM: TileMatmulFamily, LRF: ReaderFamily, RRF: ReaderFamily> StageMatmulFamily for PlaneMatmulFamily<TMM, LRF, RRF>

Source§

type LhsReader = LRF

Reader family for Lhs
Source§

type RhsReader = RRF

Reader family for Rhs
Source§

type Matmul<MP: MatmulPrecision, TL: TilingLayout, TR: TilingLayout> = PartitionedStageMatmul<MP, <TMM as TileMatmulFamily>::Matmul<MP>, <LRF as ReaderFamily>::Reader<<MP as MatmulPrecision>::ES, TL>, <RRF as ReaderFamily>::Reader<<MP as MatmulPrecision>::ES, TR>, PlanePartitioner, PlanePartitionedStageConfig<<<TMM as TileMatmulFamily>::Matmul<MP> as TileMatmul<MP>>::Config>>

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

type Config = PlanePartitionedStageConfig<<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, LRF, RRF> Freeze for PlaneMatmulFamily<TMM, LRF, RRF>

§

impl<TMM, LRF, RRF> RefUnwindSafe for PlaneMatmulFamily<TMM, LRF, RRF>
where TMM: RefUnwindSafe, LRF: RefUnwindSafe, RRF: RefUnwindSafe,

§

impl<TMM, LRF, RRF> Send for PlaneMatmulFamily<TMM, LRF, RRF>

§

impl<TMM, LRF, RRF> Sync for PlaneMatmulFamily<TMM, LRF, RRF>

§

impl<TMM, LRF, RRF> Unpin for PlaneMatmulFamily<TMM, LRF, RRF>
where TMM: Unpin, LRF: Unpin, RRF: Unpin,

§

impl<TMM, LRF, RRF> UnwindSafe for PlaneMatmulFamily<TMM, LRF, RRF>
where TMM: UnwindSafe, LRF: UnwindSafe, RRF: 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