Skip to content

[Gluon] Include bases in TensorMemoryLinearLayout mangling - #11201

Open
39ali wants to merge 2 commits into
triton-lang:mainfrom
39ali:fix-tmem-linear-layout-mangle
Open

[Gluon] Include bases in TensorMemoryLinearLayout mangling#11201
39ali wants to merge 2 commits into
triton-lang:mainfrom
39ali:fix-tmem-linear-layout-mangle

Conversation

@39ali

@39ali 39ali commented Aug 7, 2026

Copy link
Copy Markdown
Contributor

When a Gluon function is called with a layout as a compile time argument, the
compiler builds one copy of that function per layout, and tells the copies
apart by a name derived from the layout.,.

For TMEM linear layouts, the mangled name was built from the layout's shape only
but two layouts can have the same shape and still map data to memory
completely differently (different bases). Since both got the same name, the
compiler thought it had already compiled the function and reused the copy
built for the first layout. The second call silently ran code specialized
for the wrong layout .

The fix includes the bases in the name, so different layouts always get
different names and their own copies.

New contributor declaration

  • I am not making a trivial change, such as fixing a typo in a comment.

  • I have written a PR description following these
    rules.

  • I have run pre-commit run --from-ref origin/main --to-ref HEAD.

  • Select one of the following.

    • I have added tests.
      • /test for lit tests
      • /unittest for C++ tests
      • /python/test for end-to-end tests
    • This PR does not need a test because FILL THIS IN.
  • Select one of the following.

    • I have not added any lit tests.
    • The lit tests I have added follow these best practices,
      including the "tests should be minimal" section. (Usually running Python code
      and using the instructions it generates is not minimal.)

@39ali
39ali requested a review from peterbell10 as a code owner August 7, 2026 00:04
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