'tt'方言

MLIR中的Triton IR

Triton方言。

依赖的方言：

  • 算术运算:

    • addf, addi, andi, cmpf, cmpi, divf, fptosi, …

  • 数学：

    • exp, sin, cos, log, …

  • 结构化控制流:

    • for循环、if条件判断、while循环、yield生成器、condition条件

  • ControlFlow:

    • br, cond_br

[目录]

类型约束

ptr

ptr

类型

PointerType

Triton IR类型系统中的指针类型(::mlir::triton::PointerType)

Triton IR类型系统中的指针类型，可以指向标量或张量。

参数:

参数

C++类型

描述

指针指向类型

类型

addressSpace

int

TensorDesc类型

Triton IR类型系统中的张量描述符类型 (::mlir::triton::TensorDescType)

语法：

!tt.tensordesc<
  RankedTensorType   # blockType
>

一个用于nvidia-TMA描述符的可移植抽象层。

参数:

参数

C++ 类型

描述

blockType

RankedTensorType