Struct LoadSpecializationConfig

pub struct LoadSpecializationConfig {
    pub lhs: SpecializationTensorConfig,
    pub rhs: SpecializationTensorConfig,
}

Configuration for how each input tensor (Lhs and Rhs) is loaded, specifying the plane roles responsible for loading them.

Fields

lhs: SpecializationTensorConfig

Load strategy for the Lhs tensor.

rhs: SpecializationTensorConfig

Load strategy for the Rhs tensor.

Implementations

impl LoadSpecializationConfig

pub fn has_specialization(&self) -> bool

Whether there is specialization in the algorithm

impl LoadSpecializationConfig

pub fn to_plane_roles( &self, main_flow: u32, loader_tasks: MaxLoaderPlanes, ) -> PlaneRoles

Computes how many planes of each role there should be, using the number of planes needed for main execution, and how many planes each loader can handle

The strategy is to find a balanced divisor for loader planes that stays as close as possible to the main execution plane count.

Trait Implementations

impl Clone for LoadSpecializationConfig

fn clone(&self) -> LoadSpecializationConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LoadSpecializationConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for LoadSpecializationConfig

fn default() -> LoadSpecializationConfig

Returns the “default value” for a type.

impl From<LoadSpecializationConfig> for SpecializedLoadingSides

fn from(lsc: LoadSpecializationConfig) -> Self

Converts to this type from the input type.

impl Hash for LoadSpecializationConfig

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for LoadSpecializationConfig

fn eq(&self, other: &LoadSpecializationConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for LoadSpecializationConfig

impl Eq for LoadSpecializationConfig

impl StructuralPartialEq for LoadSpecializationConfig