Trait BatchConfig

Source
pub trait BatchConfig:
    Copy
    + Clone
    + Eq
    + PartialEq
    + Hash
    + Debug
    + Send
    + Sync
    + 'static {
    type GlobalConfig: GlobalConfig;

    // Required methods
    fn global_config(&self) -> Self::GlobalConfig;
    fn quantized(&self) -> bool;
    fn cube_dim(&self) -> CubeDim;
    fn line_sizes(&self) -> MatmulLineSizes;
    fn hypercube_config(&self) -> HypercubeConfig;
    fn can_yield_extra_cubes(&self) -> bool;

    // Provided method
    fn tiling_scheme(&self) -> TilingScheme { ... }
}
Expand description

Configuration for the batch matmul level.

Required Associated Types§

Source

type GlobalConfig: GlobalConfig

Underlying Global matmul config

Required Methods§

Source

fn global_config(&self) -> Self::GlobalConfig

Convert itself to the underlying global matmul config

Source

fn quantized(&self) -> bool

Returns true if the matmul is quantized.

Source

fn cube_dim(&self) -> CubeDim

Returns the CubeDim

Source

fn line_sizes(&self) -> MatmulLineSizes

Returns the line sizes for Lhs, Rhs and output

Source

fn hypercube_config(&self) -> HypercubeConfig

Returns the HypercubeConfig

Source

fn can_yield_extra_cubes(&self) -> bool

Whether it may launch more cubes than the minimum required

Provided Methods§

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§