Enum CutlassDtype 

pub enum CutlassDtype {
    F32,
    F64,
    F16,
    Bf16,
    F8E4m3,
    F8E5m2,
    F4E2m1,
    I8,
    I32,
    U8,
}

CUTLASS-side dtype tag. Each variant maps 1-to-1 onto a concrete CUTLASS C++ scalar type via CutlassDtype::as_cutlass_type.

Variants

F32

float / cutlass::float32_t.

F64

double / cutlass::float64_t.

F16

cutlass::half_t.

Bf16

cutlass::bfloat16_t.

F8E4m3

cutlass::float_e4m3_t (Hopper / Blackwell fp8).

F8E5m2

cutlass::float_e5m2_t (Hopper / Blackwell fp8).

F4E2m1

cutlass::float_e2m1_t (Blackwell fp4).

I8

int8_t (CUTLASS quantized GEMM lane).

I32

int32_t (accumulator).

U8

uint8_t.

Implementations

impl CutlassDtype

pub fn as_cutlass_type(self) -> &'static str

CUTLASS C++ type spelling for use inside a generated template instantiation. Used by the .cu source emitter.

pub fn short_name(self) -> &'static str

Stable short name used in plan-cache keys and log output.

pub fn size_bits(self) -> u32

Element size in bits. fp4 is the only sub-byte dtype.

Trait Implementations

impl Clone for CutlassDtype

fn clone(&self) -> CutlassDtype

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CutlassDtype

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

Formats the value using the given formatter.

impl Display for CutlassDtype

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

Formats the value using the given formatter.

impl Hash for CutlassDtype

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 CutlassDtype

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

impl Eq for CutlassDtype

impl StructuralPartialEq for CutlassDtype