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]Enable EmitIndicesTest #3167

Merged
merged 2 commits into from
Mar 11, 2024
Merged

Conversation

joviliast
Copy link
Contributor

Allows to test emmitting indices from different layouts. Valid also for AMD target device.

@joviliast
Copy link
Contributor Author

This PR is created to discuss a proper way of enabling EmitIndicesTest for AMD target

Comment on lines 75 to 79
#ifdef USE_ROCM
amdHelper.emitIndices(
#else
emitIndices(
#endif
Copy link
Collaborator

Choose a reason for hiding this comment

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

that kind of code is not going to be maintainable. Can you refactor this code to layer things properly, either pass different parameters or use inheritance to be able to control to call different backends.

Copy link
Contributor Author

@joviliast joviliast Mar 1, 2024

Choose a reason for hiding this comment

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

@ThomasRaoux , Is #ifdef USE_ROCM a red flag to decline merging? should I get rid of them?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

So, using #ifdef USE_ROCM is a definitely a wrong way. I've introduced new directives and build different targets for Nvidia and AMD.
Unfortunately some of the tests are failed, so I disabled them. Please take a look

@joviliast joviliast force-pushed the emitidx-test branch 4 times, most recently from 145f56f to b1a617e Compare February 23, 2024 15:43
@joviliast joviliast force-pushed the emitidx-test branch 2 times, most recently from 842d5c4 to ebd2477 Compare March 1, 2024 19:13
@joviliast joviliast force-pushed the emitidx-test branch 3 times, most recently from 2f84a39 to c475830 Compare March 8, 2024 13:17
@joviliast joviliast marked this pull request as ready for review March 8, 2024 13:21
@joviliast joviliast requested a review from ptillet as a code owner March 8, 2024 13:21
Allows to test emmitting indices from different layouts.
Valid also for AMD target device.

Signed-off-by: joviliast <iveselov.nn@gmail.com>
@zhanglx13 zhanglx13 merged commit d04f288 into triton-lang:main Mar 11, 2024
4 checks passed
htyu pushed a commit to htyu/triton that referenced this pull request Mar 20, 2024
Allows to test emmitting indices from different layouts. Valid also for
AMD target device.

Signed-off-by: joviliast <iveselov.nn@gmail.com>
karupayun pushed a commit to openxla/triton that referenced this pull request Apr 3, 2024
Allows to test emmitting indices from different layouts. Valid also for
AMD target device.

Signed-off-by: joviliast <iveselov.nn@gmail.com>
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.

4 participants