Mojo struct
UMMAKind
@register_passable(trivial)
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,
Movable,
Stringable,
UnknownDestructibility,
Writable
Aliases
KIND_F16
alias KIND_F16 = UMMAKind(__init__[__mlir_type.!pop.int_literal](2))
f16 type
KIND_F8F6F4
alias KIND_F8F6F4 = UMMAKind(__init__[__mlir_type.!pop.int_literal](3))
f8f6f4 type
KIND_I8
alias KIND_I8 = UMMAKind(__init__[__mlir_type.!pop.int_literal](4))
i8 type
KIND_TF32
alias KIND_TF32 = UMMAKind(__init__[__mlir_type.!pop.int_literal](0))
tf32 type
Methods
__eq__
__eq__(self, other: Self) -> Bool
Check if two UMMA kinds are equal.
Args:
- other (
Self): The other UMMA kind to compare with.
Returns:
True if the UMMA kinds are equal, False otherwise.
__ne__
__ne__(self, other: Self) -> Bool
Check if two UMMA kinds are not equal.
Args:
- other (
Self): The other UMMA kind to compare with.
Returns:
True if the UMMA kinds are not equal, False otherwise.
__int__
__int__(self) -> Int
Convert UMMA kind to an integer value.
Returns:
The integer value representing the UMMA instruction type.
__str__
__str__(self) -> String
Convert UMMA kind to a string, this can be used as the instruction qualifier.
Returns:
The PTX qualifier representation of the UMMA kind.
write_to
write_to[W: Writer](self, mut writer: W)
Write the UMMA kind to a writer.
Parameters:
- W (
Writer): The writer type that will receive the formatted output.
Args:
- writer (
W): 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!