Skip to content

[Gluon] Expose TMA atomic ops#10040

Merged
lezcano merged 3 commits into
mainfrom
tma_reduce_gluon
Apr 15, 2026
Merged

[Gluon] Expose TMA atomic ops#10040
lezcano merged 3 commits into
mainfrom
tma_reduce_gluon

Conversation

@lezcano
Copy link
Copy Markdown
Contributor

@lezcano lezcano commented Apr 15, 2026

tbh, I think this is also the way we should represent this in our IR.

I will add an end-to-end test once #9977 is merged (I will extend the test in that PR to test this op)

tbh, I think this is also the way we should represent this in our IR,
but well.
@lezcano lezcano requested a review from peterbell10 as a code owner April 15, 2026 10:46
Copy link
Copy Markdown
Contributor

@peterbell10 peterbell10 left a comment

Choose a reason for hiding this comment

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

Care to present an argument why this is better?

@peterbell10
Copy link
Copy Markdown
Contributor

Generally my preference is for operations that do a narowly scoped thing, over bloated interfaces with lots of flags

Copy link
Copy Markdown
Contributor

@peterbell10 peterbell10 left a comment

Choose a reason for hiding this comment

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

LGTM other than a small NIT

Comment thread python/triton/experimental/gluon/language/nvidia/hopper/tma.py
@peterbell10 peterbell10 changed the title [Gluon] Expose TMAReduction via an optional enum. [Gluon] Expose TMA atomic ops Apr 15, 2026
@lezcano lezcano enabled auto-merge (squash) April 15, 2026 13:38
@lezcano lezcano merged commit ab1f012 into main Apr 15, 2026
16 of 18 checks passed
@lezcano lezcano deleted the tma_reduce_gluon branch April 15, 2026 16:25
raymondtay pushed a commit to raymondtay/triton that referenced this pull request Apr 18, 2026
tbh, I think this is also the way we should represent this in our IR.

I will add an end-to-end test once
triton-lang#9977 is merged (I will extend
the test in that PR to test this op)
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.

2 participants