Skip to content

Update triton and triton-shared to latest upstream#10

Merged
erwei-xilinx merged 1 commit into
mainfrom
update-triton-shared-and-triton-versions
Mar 2, 2026
Merged

Update triton and triton-shared to latest upstream#10
erwei-xilinx merged 1 commit into
mainfrom
update-triton-shared-and-triton-versions

Conversation

@erwei-xilinx

Copy link
Copy Markdown
Collaborator

Summary

LLVM version note

Triton now builds against LLVM 979132a0 while mlir-aie/mlir-air wheels remain on LLVM ebf5d9ef. This works because they are separate binaries that exchange data via textual MLIR IR (stable Linalg/memref dialects). Verified end-to-end on hardware.

Important build note: if you have a cached LLVM in ~/.triton/llvm/, clear it before building (rm -rf ~/.triton/llvm/) to ensure the new LLVM version is downloaded.

Test plan

  • triton.patch applies cleanly on new triton commit
  • triton_shared.patch regenerated via rebase (1 conflict in triton-hash.txt, resolved)
  • Patch applies cleanly on clean aff7421
  • Build succeeds (Option 2: pip with --find-links)
  • Matmul example runs end-to-end on NPU hardware (AIE2)
  • Results consistent with previous versions (~2.7% BF16 numerical variance, pre-existing)

🤖 Generated with Claude Code

Update triton submodule from dbfbc1e1e (Dec 2025) to 32b7b208d
(Feb 27, 2026) and triton_shared from 209a064 to aff7421, which
bumps triton-shared's triton pin to match. This brings in 437
triton commits and a new LLVM version (979132a0, up from ebf5d9ef).

Regenerate triton_shared.patch — reduced from 4787 to 1393 lines
as upstream absorbed most of our modifications. Only one conflict
(triton-hash.txt) during rebase, resolved to match our triton
submodule pointer.

The mlir-aie/mlir-air wheels (built against LLVM ebf5d9ef) remain
compatible since they exchange textual MLIR IR with triton-shared-opt
using stable Linalg/memref dialects.

Tested: full build + matmul on NPU hardware (AIE2) with the LLVM
version divergence — identical results to previous versions.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@erwei-xilinx erwei-xilinx merged commit f5a2c69 into main Mar 2, 2026
8 of 9 checks passed
@erwei-xilinx erwei-xilinx deleted the update-triton-shared-and-triton-versions branch March 2, 2026 20:41
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.

1 participant