StagePartitioner

cubecl_matmul::components::stage

Trait StagePartitioner 

Source 
pub trait StagePartitioner:
    Send
    + Sync
    + 'static {
    // Required methods
    fn coordinates<S: StageConfig>(config: S) -> Coords2d;
    fn __expand_coordinates<S: StageConfig>(
        scope: &mut Scope,
        config: S,
    ) -> <Coords2d as CubeType>::ExpandType;
}
Expand description

Defines how the stage is partitioned among compute primitives (e.g., units or planes). Controls global writeback and and compute indexing.

Required Methods§

Source

fn coordinates<S: StageConfig>(config: S) -> Coords2d

Returns the (row, col) of the current compute primitive within the stage.

Source

fn __expand_coordinates<S: StageConfig>( scope: &mut Scope, config: S, ) -> <Coords2d as CubeType>::ExpandType

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§