Triton - Programming language

Triton is a programming language created in 2021 by Philippe Tillet.

#377on PLDB

3Years Old

Download source code:

git clone https://github.com/openai/triton

Triton is a language and compiler for parallel programming. It aims to provide a Python-based programming environment for productively writing custom DNN compute kernels capable of running at maximal throughput on modern GPU hardware.

Tags: programming language
Triton is developed on GitHub and has 12,112 stars
Early development of Triton happened in OpenAI
Triton is written in C++, Python, CMake, Markdown, YAML, reStructuredText, Bourne shell, C, Dockerfile, SVG, TOML, JSON, HTML, LLVM IR, Make
See also: (1 related languages) Numba
Read more about Triton on the web: 1.

Example from the web:

# This is a GPU kernel in Triton.
# Different instances of this
# function may run in parallel.
@jit
def add(X, Y, Z, N):
   # In Triton, each kernel instance
   # executes block operations on a
   # single thread: there is no construct
   # analogous to threadIdx
   pid = program_id(0)
   # block of indices
   idx = pid * BLOCK + arange(BLOCK)
   mask = idx < N
   # Triton uses pointer arithmetics
   # rather than indexing operators
   x = load(X + idx, mask=mask)
   y = load(Y + idx, mask=mask)
   store(Z + idx, x + y, mask=mask)