cubecl_matmul::components::batch

Struct PartitionedBatchMatmulFamily

Source 
pub struct PartitionedBatchMatmulFamily<GMM: GlobalMatmulFamily, S: GlobalPartitionMatmul> { /* private fields */ }
Expand description

Simple partitioned batch matmul family for any precision

Trait Implementations§

Source§

impl<GMM: GlobalMatmulFamily, S: GlobalPartitionMatmul> BatchMatmulFamily for PartitionedBatchMatmulFamily<GMM, S>

Source§

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

The specific BatchMatmul implementation associated with this family.
Source§

type Config = PartitionedBatchConfig<<GMM as GlobalMatmulFamily>::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, ) -> Result<Self::Config, MatmulSetupError>

Constructs the configuration based on the matmul problem, selection, and line sizes. Read more
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>, output: OutputRuntimeArg<'a, MS, R>, cube_count_input: CubeCountInputArgs<'a, R>, config: Self::Config, )

Entry point 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<GMM, S> Freeze for PartitionedBatchMatmulFamily<GMM, S>

§

impl<GMM, S> RefUnwindSafe for PartitionedBatchMatmulFamily<GMM, S>
where GMM: RefUnwindSafe, S: RefUnwindSafe,

§

impl<GMM, S> Send for PartitionedBatchMatmulFamily<GMM, S>

§

impl<GMM, S> Sync for PartitionedBatchMatmulFamily<GMM, S>

§

impl<GMM, S> Unpin for PartitionedBatchMatmulFamily<GMM, S>
where GMM: Unpin, S: Unpin,

§

impl<GMM, S> UnwindSafe for PartitionedBatchMatmulFamily<GMM, S>
where GMM: UnwindSafe, S: 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