Mojo struct
TMADescriptor
struct TMADescriptor
TMA tensor map descriptor.
An opaque 128-byte descriptor that encodes all parameters for a TMA operation,
including tensor dimensions, strides, data type, swizzle mode, and other
configuration. This descriptor is created on the host using create_tma_descriptor()
and can be passed to device code for use with TMA hardware instructions.
The descriptor must be 64-byte aligned both on the host and device.
Fields
- data (
StaticTuple[UInt8, 128]): The opaque 128-byte descriptor data.
Implemented traits
AnyType,
Copyable,
DevicePassable,
ImplicitlyCopyable,
UnknownDestructibility
Aliases
__copyinit__is_trivial
alias __copyinit__is_trivial = False
__del__is_trivial
alias __del__is_trivial = True
device_type
alias device_type = TMADescriptor
Methods
__init__
__init__(out self)
Initializes an empty TMA descriptor.
The descriptor data is uninitialized and must be filled using
create_tma_descriptor() before use.
__copyinit__
__copyinit__(out self, other: Self)
Creates a copy of a TMA descriptor.
Args:
- other (
Self): The descriptor to copy.
get_type_name
static get_type_name() -> String
Gets the type name for this descriptor.
Returns:
String: The string "TMADescriptor".
get_device_type_name
static get_device_type_name() -> String
Gets the device type name for this descriptor.
Returns:
String: The string "TMADescriptor".
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!