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_plugin 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-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-py3-none-manylinux_2_34_x86_64.whl.

File metadata

File hashes

Hashes for triton_utlx-3.7.0-py3-none-manylinux_2_34_x86_64.whl
Algorithm Hash digest
SHA256 8c54c79c5508c668760f6b70fb2c0e27bc69fd4830b2c46b4bc7cfcc36c2bf9b
MD5 cfa7ad31230badf154edff1d0a8c5df6
BLAKE2b-256 8d0c34156cb3360e438e4904be182b7776fa680e217ba357f4724f0c4b1b0e3a

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