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

[AMD][Navi31] Convert WMMA dot op to LLVM #3199

Merged
merged 2 commits into from
Mar 5, 2024

Conversation

joviliast
Copy link
Contributor

@joviliast joviliast commented Feb 26, 2024

-Add WMMA conversion logic for dot operation
-Fix helper hunctions for WMMA layout
-Add lit test for WMMA dot operation conversion

* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#ifdef USE_ROCM
Copy link
Collaborator

Choose a reason for hiding this comment

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

can we remove this?


} // namespace

LogicalResult convertWMMA(triton::DotOp op, triton::DotOp::Adaptor adaptor,
Copy link
Collaborator

Choose a reason for hiding this comment

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

can you add a basic lit tests to make sure this codes runs.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Yep. working on it

@joviliast joviliast force-pushed the wmma-convert-dot branch 3 times, most recently from b3843b3 to 66e7ea7 Compare February 29, 2024 16:17
@joviliast joviliast marked this pull request as draft February 29, 2024 16:18
@joviliast
Copy link
Contributor Author

#3254 - this pr is also required for testing

@joviliast
Copy link
Contributor Author

Added lit test. Please check.
Will convert PR from draft after #3254 merged.

@joviliast joviliast requested a review from ThomasRaoux March 1, 2024 18:47
@joviliast joviliast marked this pull request as ready for review March 5, 2024 14:49
Copy link
Collaborator

@ThomasRaoux ThomasRaoux left a comment

Choose a reason for hiding this comment

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

LGTM

-Add WMMA conversion logic for dot operation
-Fix helper hunctions for WMMA layout
-Add lit test for WMMA dot operation conversion

Signed-off-by: joviliast <iveselov.nn@gmail.com>
@ThomasRaoux ThomasRaoux enabled auto-merge (squash) March 5, 2024 18:03
@ThomasRaoux ThomasRaoux disabled auto-merge March 5, 2024 18:03
@ThomasRaoux ThomasRaoux enabled auto-merge (squash) March 5, 2024 18:03
@ThomasRaoux ThomasRaoux merged commit 9cfad37 into triton-lang:main Mar 5, 2024
4 checks passed
ThomasRaoux added a commit to ThomasRaoux/triton that referenced this pull request Mar 5, 2024
@ThomasRaoux
Copy link
Collaborator

sorry reverting this as it seems to cause a build failure due to a previous PR. Could you resubmit it?

ThomasRaoux added a commit that referenced this pull request Mar 5, 2024
This reverts commit 9cfad37.


There has been an inflight collision with some refactoring causing a
build break
@joviliast
Copy link
Contributor Author

sorry reverting this as it seems to cause a build failure due to a previous PR. Could you resubmit it?

#3292 done

binarman pushed a commit to binarman/triton that referenced this pull request Apr 2, 2024
-Add WMMA conversion logic for dot operation
-Fix helper hunctions for WMMA layout
-Add lit test for WMMA dot operation conversion

Signed-off-by: joviliast <iveselov.nn@gmail.com>
binarman pushed a commit to binarman/triton that referenced this pull request Apr 2, 2024
…triton-lang#3284)

This reverts commit 52b3549.


There has been an inflight collision with some refactoring causing a
build break
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