Struct ScaleSet 

pub struct ScaleSet {
    pub a: Option<*const f32>,
    pub b: Option<*const f32>,
    pub c: Option<*const f32>,
    pub d: Option<*const f32>,
}

Bundle of optional scale pointers for cuBLASLt fp8 matmul.

Each pointer is either:

• None (omit the attribute — cuBLASLt assumes scale 1.0),
• Some(ptr) where ptr is a device pointer to one or more f32 scale values. For per-tensor scale supply a single f32; for per-row scale supply m (or n) f32s in row-major layout.

Fields

a: Option<*const f32>

b: Option<*const f32>

c: Option<*const f32>

d: Option<*const f32>

Implementations

impl ScaleSet

pub const fn empty() -> Self

pub fn with_a(self, ptr: *const f32) -> Self

pub fn with_b(self, ptr: *const f32) -> Self

pub fn with_c(self, ptr: *const f32) -> Self

pub fn with_d(self, ptr: *const f32) -> Self

pub fn is_empty(&self) -> bool

pub unsafe fn apply(&self, desc: cublasLtMatmulDesc_t) -> Result<(), String>

Write each Some(ptr) onto the descriptor. Returns the first error encountered, if any.

Safety

desc must be a live cublasLtMatmulDesc_t. The scale pointers must remain valid for the entire lifetime of any matmul call that uses desc.

Trait Implementations

impl Clone for ScaleSet

fn clone(&self) -> ScaleSet

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ScaleSet

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

Formats the value using the given formatter.

impl Default for ScaleSet

fn default() -> ScaleSet

Returns the “default value” for a type.

impl Copy for ScaleSet

impl Send for ScaleSet

impl Sync for ScaleSet