Enum TensorMapDataType 

pub enum TensorMapDataType {
    UInt8,
    UInt16,
    UInt32,
    Int32,
    UInt64,
    Int64,
    Float16,
    Float32,
    Float64,
    BFloat16,
    Float32Ftz,
    TFloat32,
    TFloat32Ftz,
    Float8E4m3,
    Float8E5m2,
    Float4E2m1,
    Float6E2m3,
    Float6E3m2,
}

Element dtype consumed by the TMA. Matches [cudarc::driver::sys::CUtensorMapDataType_enum] one-to-one. We duplicate the enum so callers don’t depend on cudarc’s sys module directly (which is gated on a CUDA-version feature in cudarc).

Variants

UInt8

UInt16

UInt32

Int32

UInt64

Int64

Float16

Float32

Float64

BFloat16

Float32Ftz

TFloat32

TF32 / __nv_tf32.

TFloat32Ftz

Float8E4m3

Blackwell-only fp4/fp6/fp8 variants. Available with the blackwell cargo feature only — the host-side enum is allowed regardless of feature so unit tests can round-trip the value; runtime kernels need a Blackwell driver.

Float8E5m2

Float4E2m1

Float6E2m3

Float6E3m2

Implementations

impl TensorMapDataType

pub fn as_u32(self) -> u32

Numeric value used by the underlying CUDA driver. Matches the public ABI of CUtensorMapDataType_enum.

Trait Implementations

impl Clone for TensorMapDataType

fn clone(&self) -> TensorMapDataType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorMapDataType

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

Formats the value using the given formatter.

impl PartialEq for TensorMapDataType

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

impl Eq for TensorMapDataType

impl StructuralPartialEq for TensorMapDataType