‘tt’ Dialect

The Triton IR in MLIR

Triton Dialect.

Dependent Dialects:

• Arith:

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

• Math:

  • exp, sin, cos, log, …

• StructuredControlFlow:

  • for, if, while, yield, condition

• ControlFlow:

  • br, cond_br

[TOC]

Type constraints

ptr

ptr

Types

PointerType

Pointer type (::mlir::triton::PointerType) in Triton IR type system

Pointer type in Triton IR type system, which could be pointing to scalars or tensors.

Parameters:

Parameter	C++ type	Description
pointeeType	Type
addressSpace	int

TensorDescType

Tensor descriptor type (::mlir::triton::TensorDescType) in Triton IR type system

Syntax:

!tt.tensordesc<
  RankedTensorType   # blockType
>

A portable abstraction for nvidia-TMA descriptors.

Parameters:

Parameter	C++ type	Description
blockType	RankedTensorType