-
Notifications
You must be signed in to change notification settings - Fork 1.6k
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
[AMD] Reland instruction scheduling hint changes #4940
Open
ravil-mobile
wants to merge
7
commits into
triton-lang:main
Choose a base branch
from
ravil-mobile:ravil/bug-fix
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
+647
−120
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
from
October 17, 2024 09:56
48cc87a
to
dfcb55e
Compare
ravil-mobile
changed the title
Ravil/bug fix
[AMD] Fixed a bug resulted in reverting PR#4919
Oct 17, 2024
7 tasks
antiagainst
changed the title
[AMD] Fixed a bug resulted in reverting PR#4919
[AMD] Reland instruction scheduling hint changes
Oct 17, 2024
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
2 times, most recently
from
October 18, 2024 09:57
5b044e9
to
73b15e8
Compare
ravil-mobile
requested review from
antiagainst,
zhanglx13 and
ptillet
as code owners
October 18, 2024 09:58
giuseros
reviewed
Oct 18, 2024
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
2 times, most recently
from
October 22, 2024 10:59
4cb27d1
to
00ab1fe
Compare
This reverts commit 93de426.
Replaced temlate-based impl. of `rewindUnaryOps` in `SchedInstructions.cpp` using regular for-loops. The new impl. is more robust and can handle other unary ops automatically.
giuseros
reviewed
Oct 22, 2024
third_party/amd/lib/TritonAMDGPUTransforms/StreamPipelineV2.cpp
Outdated
Show resolved
Hide resolved
* add a test for the presence of OpIdx attribute
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
from
October 22, 2024 12:54
00ab1fe
to
97f0e1e
Compare
The extra check tests whether the data are loaded from HBM using `buffer_load` instructions. The CKV3 scheduling is skipped if the check fails.
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
from
October 22, 2024 13:30
f6f901e
to
7b74c32
Compare
ravil-mobile
force-pushed
the
ravil/bug-fix
branch
from
October 23, 2024 13:31
1b3ef15
to
8a32b31
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This commit relands #4819
with the following fixes:
rewindUnaryOps
to use regular for-loops. The new way is more robust and can handle other unary ops automatically.instr.sched.barriers
using the ones fromrocdl
dialect from the MLIR upstream