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

Refactor existing MMA intrinsics #19098

Merged
merged 2 commits into from
Nov 12, 2024
Merged

Conversation

bjacob
Copy link
Contributor

@bjacob bjacob commented Nov 11, 2024

Some code simplifications in IREEGPUAttrs.cpp, and some shuffling of the MMAIntrinsic enum, making it easier to add more MMAIntrinsics in the future (see #19099).

@bjacob bjacob changed the base branch from main to users/bjacob/f64 November 11, 2024 15:56
@bjacob bjacob force-pushed the users/bjacob/refactor-mma-intrinsics branch from 638d50f to a44474a Compare November 11, 2024 15:57
@bjacob bjacob marked this pull request as ready for review November 11, 2024 16:50
Copy link
Member

@kuhar kuhar left a comment

Choose a reason for hiding this comment

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

The new enum encoding format makes sense to me

Signed-off-by: Benoit Jacob <[email protected]>
Signed-off-by: Benoit Jacob <[email protected]>
@bjacob bjacob requested a review from benvanik as a code owner November 11, 2024 21:57
@bjacob bjacob force-pushed the users/bjacob/refactor-mma-intrinsics branch from a44474a to 17dc529 Compare November 11, 2024 21:57
@raikonenfnu raikonenfnu self-requested a review November 12, 2024 00:13
Copy link
Collaborator

@raikonenfnu raikonenfnu left a comment

Choose a reason for hiding this comment

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

Super awesome cleanup! Thanks Benoit! :)

Base automatically changed from users/bjacob/f64 to main November 12, 2024 15:21
@bjacob bjacob merged commit bbb87aa into main Nov 12, 2024
36 checks passed
@bjacob bjacob deleted the users/bjacob/refactor-mma-intrinsics branch November 12, 2024 16:20
raikonenfnu added a commit to raikonenfnu/iree that referenced this pull request Nov 14, 2024
Apply similar cleanups to what is done in iree-org#19098.
For most part we do:
1. Templateize getVectorType and getOpaqueMmaLayout to work on any
   intrinsic
2. Use common getOpaqueMmaLayout for VirtualMMA
3. Update getABCElementTypes to be similar to MMAAttr

Signed-off-by: Stanley Winata <[email protected]>
raikonenfnu added a commit to raikonenfnu/iree that referenced this pull request Nov 14, 2024
Apply similar cleanups to what is done in iree-org#19098.
For most part we do:
1. Templateize getVectorType and getOpaqueMmaLayout to work on any
   intrinsic
2. Use common getOpaqueMmaLayout for VirtualMMA
3. Update getABCElementTypes to be similar to MMAAttr
4. Rename get*MFMA fn to get*MMA since MFMA is CDNA specific but in
   reality it does not have to be MFMA instructions
Signed-off-by: Stanley Winata <[email protected]>

Signed-off-by: Stanley Winata <[email protected]>
bjacob pushed a commit that referenced this pull request Nov 14, 2024
…MAAttr (#19144)

Apply similar cleanups to what is done in
#19098. For most part we do:
1. Templateize getVectorType and getOpaqueMmaLayout to work on any
intrinsic
2. Use common getOpaqueMmaLayout for VirtualMMA
3. Update getABCElementTypes to be similar to MMAAttr
4. Rename get.*MFMA fn to get.*MMA since MFMA is CDNA specific but in
   reality it does not have to be MFMA instructions

Signed-off-by: Stanley Winata <[email protected]>
Groverkss pushed a commit to Groverkss/iree that referenced this pull request Dec 1, 2024
Some code simplifications in `IREEGPUAttrs.cpp`, and some shuffling of
the MMAIntrinsic enum, making it easier to add more MMAIntrinsics in the
future (see iree-org#19099).

---------

Signed-off-by: Benoit Jacob <[email protected]>
Groverkss pushed a commit to Groverkss/iree that referenced this pull request Dec 1, 2024
…MAAttr (iree-org#19144)

Apply similar cleanups to what is done in
iree-org#19098. For most part we do:
1. Templateize getVectorType and getOpaqueMmaLayout to work on any
intrinsic
2. Use common getOpaqueMmaLayout for VirtualMMA
3. Update getABCElementTypes to be similar to MMAAttr
4. Rename get.*MFMA fn to get.*MMA since MFMA is CDNA specific but in
   reality it does not have to be MFMA instructions

Signed-off-by: Stanley Winata <[email protected]>
giacs-epic pushed a commit to giacs-epic/iree that referenced this pull request Dec 4, 2024
Some code simplifications in `IREEGPUAttrs.cpp`, and some shuffling of
the MMAIntrinsic enum, making it easier to add more MMAIntrinsics in the
future (see iree-org#19099).

---------

Signed-off-by: Benoit Jacob <[email protected]>
Signed-off-by: Giacomo Serafini <[email protected]>
giacs-epic pushed a commit to giacs-epic/iree that referenced this pull request Dec 4, 2024
…MAAttr (iree-org#19144)

Apply similar cleanups to what is done in
iree-org#19098. For most part we do:
1. Templateize getVectorType and getOpaqueMmaLayout to work on any
intrinsic
2. Use common getOpaqueMmaLayout for VirtualMMA
3. Update getABCElementTypes to be similar to MMAAttr
4. Rename get.*MFMA fn to get.*MMA since MFMA is CDNA specific but in
   reality it does not have to be MFMA instructions

Signed-off-by: Stanley Winata <[email protected]>
Signed-off-by: Giacomo Serafini <[email protected]>
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.

3 participants