‘tt’ 方言¶
MLIR 中的 Triton IR
Triton 方言。
依赖的方言
Arith
addf, addi, andi, cmpf, cmpi, divf, fptosi, …
Math
exp, sin, cos, log, …
StructuredControlFlow
for, if, while, yield, condition
ControlFlow
br, cond_br
[目录]
类型约束¶
ptr¶
ptr¶
类型¶
PointerType¶
指针类型 (::mlir::triton::PointerType) 在 Triton IR 类型系统中
Triton IR 类型系统中的指针类型，可以指向标量或张量。
参数:¶
参数 |
C++ 类型 |
描述 |
|---|---|---|
pointeeType |
|
|
addressSpace |
|
TensorDescType¶
张量描述符类型 (::mlir::triton::TensorDescType) 在 Triton IR 类型系统中
语法
!tt.tensordesc<
RankedTensorType # blockType
>
nvidia-TMA 描述符的可移植抽象。
参数:¶
参数 |
C++ 类型 |
描述 |
|---|---|---|
blockType |
|