Top 1,000 Features Creators Events Podcasts Extensions Blog Explorer CSV Download

Triton

< >

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

#579on PLDB 3Years Old
Download source code:
git clone https://github.com/openai/triton

This is the development repository of Triton, a language and compiler for writing highly efficient custom Deep-Learning primitives. The aim of Triton is to provide an open-source environment to write fast code at higher productivity than CUDA, but also with higher flexibility than other existing DSLs.


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)

View source

- Build the next great programming language About Resources Acknowledgements Part of the World Wide Scroll