Skip to content

Triton v3.6.x iluvatar backend and 5 TLE primitives support#724

Open
Salamanca001 wants to merge 2 commits into
flagos-ai:triton_v3.6.xfrom
Salamanca001:triton_v3.6.x_iluvatar
Open

Triton v3.6.x iluvatar backend and 5 TLE primitives support#724
Salamanca001 wants to merge 2 commits into
flagos-ai:triton_v3.6.xfrom
Salamanca001:triton_v3.6.x_iluvatar

Conversation

@Salamanca001

Copy link
Copy Markdown
Contributor

This PR brings the Iluvatar backend support onto Triton 3.6 in FlagTree and adds Iluvatar TLE lowering support.
Included commits:

  • 7b4cac885 [BACKEND] update iluvatar backend support on triton3.6.
  • 13abd15d8 [TLE][ILUVATAR] Add TLE support for alloc, local_ptr, copy, extract_tile and insert_tile.
    Main changes:
  • Add third_party/iluvatar backend integration, including compiler/driver entry points, Iluvatar GPU dialect, lowering passes, target info, utility code, build wiring, and test runner.
  • Add Iluvatar TLE dialect/lowering support for alloc, local_ptr, copy, extract_tile, and insert_tile.
  • Add local pointer optimization passes, barrier insertion, TLE-to-LLVM lowering utilities, and registration into the Iluvatar build/dialect pipeline.
  • Wire Iluvatar setup/build support into top-level CMake and Python setup helpers.

@CLAassistant

CLAassistant commented Jun 26, 2026

Copy link
Copy Markdown

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

if os.environ.get("FLAGTREE_BACKEND") == "iluvatar":
triton_pkg_dir = repo_root / "third_party" / "iluvatar" / "python" / "triton"
else:
triton_pkg_dir = repo_root / "python" / "triton"

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Comment thread CMakeLists.txt
flagtree_add_tle_generated_header_dependencies()
endif()
add_subdirectory(test)
elif (FLAGTREE_BACKEND STREQUAL "iluvatar")

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

elseif(FLAGTREE_BACKEND STREQUAL "iluvatar")

@sunnycase sunnycase left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Thanks for the work here. Before this is finalized, could you please add a concise summary of the TLE primitive implementation plan?

It would be helpful to cover the main design points, such as the abstraction/lowering flow, compiler/runtime integration points, supported operator scope, dtype/shape/backend limitations, and the validation approach. Could you also include performance data for a few representative operators, ideally with baseline vs. TLE primitive numbers, test shapes, hardware/backend configuration, and measurement methodology?

For the expected level of detail and presentation style, PR #617 could be a useful reference: #617

@Salamanca001 Salamanca001 changed the title Triton v3.6.x iluvatar Triton v3.6.x iluvatar backend and 5 TLE primitives support Jun 29, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants