Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CODE SHARING] Ravil/sched inst #611

Draft
wants to merge 10 commits into
base: sjw-pipeline-infra
Choose a base branch
from

Conversation

ravil-mobile
Copy link

Implemented the insertion of the scheduling primitives to MLIR/LLVM

…structure

    - Copied scheduler from MatmulLoopPipeline (much could be consolidated)
    - Enable register buffering (even though may increases register pressure)
    - Enable num_stages=2+, including multi-buffering, and make `2` the default
    - updated tutorial for new tuning default
    - added lit	tests
- Also move independent(from loop-carried buffer) `triton_gpu.local_store` as early as possible
- check for last atomic (sync?)
- also check for other accesses to the source
@ravil-mobile ravil-mobile changed the title Ravil/sched inst [CODE SHARING] Ravil/sched inst Jul 10, 2024
@ravil-mobile ravil-mobile force-pushed the ravil/sched-inst branch 2 times, most recently from e254a02 to 1abb7b4 Compare July 10, 2024 15:05
@ravil-mobile ravil-mobile changed the base branch from main to sjw-pipeline-infra July 10, 2024 15:45
@sjw36 sjw36 force-pushed the sjw-pipeline-infra branch 2 times, most recently from 4eeb8cc to faf95cb Compare July 18, 2024 14:35
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