Modules§
Structs§
- Binary
Instruction - Binding
- Body
- A body is composed of a list of instructions.
- Compilation
Options - Compute
Kernel - Const
Array - CppCompiler
- CppSupported
Features - Cube
Index Flags - Cube indexes flags. When true the corresponding index is declared and computed as needed in the kernel.
- Flags
- Flags gathered during Cube IR translation for the kernel compilation.
- Fragment
- Index
Assign Instruction - Index
Instruction - Indexed
Variable - Item
- Local
Array - Manual
Mma - MmaShape
- Optimized
Args - Shared
Memory - Unary
Instruction
Enums§
- Atomic
Kind - Elem
- FP4Kind
- FP6Kind
- FP8Kind
- Fragment
Ident - Fragment
Layout - Instruction
- Variable
- Warp
Instruction - Wmma
Instruction - Warp Matrix-Multiply and Accumulate Instruction.
Traits§
- Architecture
- Component
- Dialect
- Dialect
Bindings - Dialect
Cube Builtins - Dialect
Includes - Dialect
Instructions - Dialect
Processors - IR Processors to be applied to the scopes during processing. [
CheckedIO] is always applied by default, so these are only for target specific processors like MMA index processors. - Dialect
Types - Dialect
Warp Reduce Compiler - Dialect
Wmma Compiler - FmtLeft
Functions§
- compile_
bindings - compile_
scalars_ dynamic - compile_
scalars_ static - frag_
as_ ptr - frag_
ident_ str - frag_
layout_ str - register_
mma_ features - register_
scaled_ mma_ features - register_
supported_ types - register_
wmma_ features - type_
definitions - type_
info_ definition - type_
scalar_ definitions - type_
vectorized_ definitions - variable_
to_ frag