Skip to content

[TLE] Move TLE distributed API under language namespace#448

Merged
sunnycase merged 2 commits intotriton_v3.3.xfrom
codex/tle-language-distributed-v33
Mar 19, 2026
Merged

[TLE] Move TLE distributed API under language namespace#448
sunnycase merged 2 commits intotriton_v3.3.xfrom
codex/tle-language-distributed-v33

Conversation

@sunnycase
Copy link
Copy Markdown
Collaborator

@sunnycase sunnycase commented Mar 19, 2026

Summary

  • move python/triton/experimental/tle/distributed.py to python/triton/experimental/tle/language/distributed.py
  • stop re-exporting language internals from triton.experimental.tle.__init__
  • re-export distributed APIs from triton.experimental.tle.language.__init__ so users can call tle.device_mesh, tle.remote, etc. via import triton.experimental.tle.language as tle
  • update TLE DSA example to use import triton.experimental.tle.language as tle and tle.dsa.*
  • update distributed API docstrings from tleg.local_ptr to tle.dsa.local_ptr

Validation

  • python -m py_compile python/triton/experimental/tle/__init__.py python/triton/experimental/tle/language/__init__.py python/triton/experimental/tle/language/distributed.py third_party/tsingmicro/examples/tle/test_tle_dsa_noc_gemm_4096.py

Notes

  • runtime import validation requiring triton._C.libtriton was not run in this environment (extension not built).

@sunnycase sunnycase requested a review from zhzhcookie as a code owner March 19, 2026 02:06
Comment thread python/triton/experimental/tle/__init__.py
@zhzhcookie zhzhcookie changed the title [triton_v3.3.x] Move TLE distributed API under language namespace [TLE] Move TLE distributed API under language namespace Mar 19, 2026
Copy link
Copy Markdown
Collaborator

@zhzhcookie zhzhcookie left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LG

@sunnycase sunnycase merged commit 3e6fc2c into triton_v3.3.x Mar 19, 2026
6 checks passed
@sunnycase sunnycase deleted the codex/tle-language-distributed-v33 branch March 19, 2026 05:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants