Skip to main content

uTLX: Triton Language Extensions distributed as a Plugin

Project description

uTLX — Triton Language Extensions (Plugin)

utlx ships a Triton plugin (libutlx.so) plus the utlx Python DSL that exposes the TLX dialect (async loads, warp-group MMA, shared/tensor-memory buffers, etc.).

Compatibility

This wheel contains a native libutlx.so that links against libtriton.so and is ABI-coupled to a specific Triton build. Build Triton from the upstream triton-lang/triton repository at the v3.7.0 tag:

git clone https://github.com/triton-lang/triton
cd triton && git checkout v3.7.0
pip install -e . --no-build-isolation

Note: at the v3.7.0 tag the plugin custom-op binding in python/src/ir.cc discards the op result, so TLX ops return None. Apply the one-line fix (prepend an empty result slot, call addOp, return args[0]) so ops return their created value.

Requirements

  • Linux x86_64, glibc >= 2.14, libstdc++ from GCC 11+ (GLIBCXX_3.4.29).
  • Set TRITON_PLUGIN_PATHS to the installed libutlx.so and build Triton with TRITON_EXT_ENABLED=ON.

License

MIT. Copyright (c) 2026 Meta Platforms, Inc. and affiliates.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

triton_utlx-3.7.0.post1-py3-none-manylinux_2_34_x86_64.whl (1.7 MB view details)

Uploaded Python 3manylinux: glibc 2.34+ x86-64

File details

Details for the file triton_utlx-3.7.0.post1-py3-none-manylinux_2_34_x86_64.whl.

File metadata

File hashes

Hashes for triton_utlx-3.7.0.post1-py3-none-manylinux_2_34_x86_64.whl
Algorithm Hash digest
SHA256 602588526344aa61b4f116fdff86ad320e1a6baef9bf865eab62d1b27da2f49c
MD5 6ace1ec8a212ae8a1f816adffe23e5fb
BLAKE2b-256 ec0fdde2b6ce5136f49c3672eae52a9574f85d2da49bef7945cdf19486559d63

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page