Skip to content
This repository has been archived by the owner on Jun 27, 2024. It is now read-only.

[Triton] num-warps and num-stages should be a property of triton.executable.export #124

Open
ezhulenev opened this issue Jun 9, 2023 · 0 comments
Open

[Triton] num-warps and num-stages should be a property of triton.executable.export #124

ezhulenev opened this issue Jun 9, 2023 · 0 comments
Labels
enhancement New feature or request

Comments

@ezhulenev
Copy link
Contributor

Today num-warps and num-stages are configured globally in properties here: https://github.com/openxla/openxla-nvgpu/blob/628fc75bcfe00c0ab2551824914a896668f21909/compiler/src/openxla/compiler/nvgpu/Dialect/TritonFlow/Conversion/ConvertTritonToFlowDispatch.h#L29-L33

It should be an attribute on triton.call (or tt.func) and triton.executable.export because different triton function can have different parameters.
@ezhulenev ezhulenev added the enhancement New feature or request label Jun 9, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ezhulenev