“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

Type

addressSpace

int

TensorDescType

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

语法

!tt.tensordesc<
  RankedTensorType   # blockType
>

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

参数:

参数

C++ 类型

描述

blockType

RankedTensorType