‘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

MemDescType

memory descriptor type (::mlir::triton::MemDescType) in Triton IR type system

Memory descriptor contains a base pointer (scalar) and a descriptor of the memory. If mutable memory is false that means the memory is constant and can only be allocated and stored once. A constant memory allocation is different than a tensor as it can have multiple views and the descriptor can be changed without changing the underlying memory.

Parameters:

Parameter

C++ type

Description

shape

::llvm::ArrayRef<int64_t>

elementType

Type

encoding

Attribute

memorySpace

Attribute

mutable_memory

bool

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