‘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
[TOC]
类型约束
ptr
类型
指针类型 (PointerType)
Triton IR 类型系统中的指针类型 (::mlir::triton::PointerType)
Triton IR 类型系统中的指针类型。指针类型只能指向标量元素类型。
参数：
参数 |
C++ 类型 |
描述 |
|---|---|---|
pointeeType |
|
|
addressSpace |
|
张量描述符类型 (TensorDescType)
张量描述符类型（::mlir::triton::TensorDescType）在 Triton IR 类型系统中
一种用于 TMA（张量内存访问）描述符的可移植抽象。这是用于分块张量内存访问的基础张量描述符类型。
Shape（形状）和 elementType（元素类型）描述了块的维度和数据类型。可选的 sharedLayout（共享布局）属性携带在降低（lowering）过程中分配的共享内存编码（例如 swizzle 模式）。
对于 im2col 等专用访问模式，请参阅 TritonNvidiaGPU 方言中的 TensorDescIm2ColType。
参数：
参数 |
C++ 类型 |
描述 |
|---|---|---|
形状 (shape) |
|
|
elementType |
|
|
sharedLayout |
|