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.
Aliases
KIND_TF32 = UMMAKind(__init__[__mlir_type.!pop.int_literal](0)): tf32 typeKIND_F16 = UMMAKind(__init__[__mlir_type.!pop.int_literal](2)): f16 typeKIND_F8F6F4 = UMMAKind(__init__[__mlir_type.!pop.int_literal](3)): f8f6f4 typeKIND_I8 = UMMAKind(__init__[__mlir_type.!pop.int_literal](4)): i8 type
Implemented traits
AnyType,
Copyable,
ExplicitlyCopyable,
Movable,
Stringable,
UnknownDestructibility,
Writable
Methods
__init__
@implicit
__init__(value: SIMD[int32, 1]) -> Self
Initialize UMMA kind with an integer value.
Args:
- value (
SIMD[int32, 1]): The integer value representing the UMMA instruction type.
__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!