Enum CuTensorMapDataType 

#[repr(u32)]
pub enum CuTensorMapDataType {
    Uint8 = 0,
    Uint16 = 1,
    Uint32 = 2,
    Int32 = 3,
    Uint64 = 4,
    Int64 = 5,
    Float16 = 6,
    Float32 = 7,
    Float64 = 8,
    Bfloat16 = 9,
    Float32Ftz = 10,
    TF32 = 11,
    TF32Ftz = 12,
}

Element data type for the TMA descriptor.

Corresponds to CUtensorMapDataType in the CUDA Driver API.

Variants

Uint8 = 0

Unsigned 8-bit integer.

Uint16 = 1

Unsigned 16-bit integer.

Uint32 = 2

Unsigned 32-bit integer.

Int32 = 3

Signed 32-bit integer.

Uint64 = 4

Unsigned 64-bit integer.

Int64 = 5

Signed 64-bit integer.

Float16 = 6

IEEE-754 half-precision float (f16).

Float32 = 7

IEEE-754 single-precision float (f32).

Float64 = 8

IEEE-754 double-precision float (f64).

Bfloat16 = 9

Brain float 16 (bfloat16).

Float32Ftz = 10

f32 with flush-to-zero for subnormals.

TF32 = 11

TensorFloat-32 (TF32).

TF32Ftz = 12

TF32 with flush-to-zero.

Implementations

impl CuTensorMapDataType

pub const fn element_size_bytes(self) -> u32

Returns the element size in bytes.

Trait Implementations

impl Clone for CuTensorMapDataType

fn clone(&self) -> CuTensorMapDataType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for CuTensorMapDataType

impl Debug for CuTensorMapDataType

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

Formats the value using the given formatter.

impl Eq for CuTensorMapDataType

impl Hash for CuTensorMapDataType

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 CuTensorMapDataType

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