For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo struct
UMMAKind
struct UMMAKind
Struct for UMMA instruction types.
This struct defines the different types of UMMA instructions that is supported by BlackWell.
Implemented traits
AnyType,
Copyable,
Equatable,
Hashable,
ImplicitlyCopyable,
ImplicitlyDeletable,
Movable,
RegisterPassable,
TrivialRegisterPassable,
Writable
comptime members
KIND_F16
comptime KIND_F16 = UMMAKind(Int32(2))
F16 type.
KIND_F8F6F4
comptime KIND_F8F6F4 = UMMAKind(Int32(3))
F8F6F4 type.
KIND_I8
comptime KIND_I8 = UMMAKind(Int32(4))
I8 type.
KIND_MXF4
comptime KIND_MXF4 = UMMAKind(Int32(6))
MXF4 type.
KIND_MXF4NVF4
comptime KIND_MXF4NVF4 = UMMAKind(Int32(7))
MXF4NVF4 type.
KIND_MXF8F6F4
comptime KIND_MXF8F6F4 = UMMAKind(Int32(5))
MXF8F6F4 type.
KIND_TF32
comptime KIND_TF32 = UMMAKind(Int32(0))
TF32 type.
Methods
__eq__
def __eq__(self, other: Self) -> Bool
Check if two UMMA kinds are equal.
Args:
- other (
Self): The other UMMA kind to compare with.
Returns:
Bool: True if the UMMA kinds are equal, False otherwise.
__ne__
def __ne__(self, other: Self) -> Bool
Check if two UMMA kinds are not equal.
Args:
- other (
Self): The other UMMA kind to compare with.
Returns:
Bool: True if the UMMA kinds are not equal, False otherwise.
__int__
def __int__(self) -> Int
Convert UMMA kind to an integer value.
Returns:
Int: The integer value representing the UMMA instruction type.
write_to
def write_to(self, mut writer: T)
Write the UMMA kind to a writer.
Args:
- writer (
T): The writer to write the UMMA kind to.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!