Skip to content

Supported OPs - Triton

This page lists the supported Triton Ops validated in the Triton execution environment.

The operations below can be used on RBLN NPUs. The supported set will expand over time. Using unsupported operations may result in unexpected compilation errors.

Reference: triton.language (Triton official documentation)

Programming Model

Triton OPs Description
tensor -

Creation Ops

Triton OPs Description
zeros -

Shape Manipulation Ops

Triton OPs Description
broadcast -
broadcast_to -
expand_dims -
permute -
reshape -
trans -

Linear Algebra Ops

Triton OPs Description
dot -

Memory/Pointer Ops

Triton OPs Description
load -
store -
make_block_ptr -
advance -

Indexing Ops

Triton OPs Description
where -

Math Ops

Triton OPs Description
abs -
ceil -
cos -
erf -
exp -
exp2 -
fdiv -
floor -
log -
log2 -
maximum -
minimum -
rsqrt -
sigmoid -
sin -
sqrt -

Reduction Ops

Triton OPs Description
max -
min -
sum -

Iterators

Triton OPs Description
static_range -

Debug Ops

Triton OPs Description
static_print -
static_assert -