Struct MixedPrecisionConfig 

pub struct MixedPrecisionConfig {
    pub storage_precision: StoragePrecision,
    pub compute_precision: ComputePrecision,
    pub algorithm: MixedSpMVAlgo,
    pub sm_version: SmVersion,
}

Configuration for mixed-precision SpMV.

Fields

storage_precision: StoragePrecision

Storage precision for matrix values (FP16 or BF16).

compute_precision: ComputePrecision

Compute/accumulation precision (FP32 for now).

algorithm: MixedSpMVAlgo

Algorithm selection.

sm_version: SmVersion

Target GPU architecture.

Implementations

impl MixedPrecisionConfig

pub const fn fp16_fp32(algo: MixedSpMVAlgo, sm: SmVersion) -> Self

Creates a new configuration with FP16 storage and FP32 accumulation.

pub const fn bf16_fp32(algo: MixedSpMVAlgo, sm: SmVersion) -> Self

Creates a new configuration with BF16 storage and FP32 accumulation.

Trait Implementations

impl Clone for MixedPrecisionConfig

fn clone(&self) -> MixedPrecisionConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MixedPrecisionConfig

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

Formats the value using the given formatter.

impl Hash for MixedPrecisionConfig

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 MixedPrecisionConfig

fn eq(&self, other: &MixedPrecisionConfig) -> 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 MixedPrecisionConfig

impl Eq for MixedPrecisionConfig

impl StructuralPartialEq for MixedPrecisionConfig