‘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

ptr

类型

指针类型 (PointerType)

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

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

参数：

参数

C++ 类型

描述

pointeeType

Type

addressSpace

int

张量描述类型 (TensorDescType)

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

语法

!tt.tensordesc<
  RankedTensorType   # blockType
>

NVIDIA TMA 描述符的可移植抽象。

参数：

参数

C++ 类型

描述

blockType

RankedTensorType