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

[ROCm][Bugfix] Ensure that the moe_wna16_gemm kernel is not built on ROCm platforms. #14629

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

SageMoore
Copy link
Contributor

@SageMoore SageMoore commented Mar 11, 2025

See above. This kernel was introduced in #13321.

Signed-off-by: Sage Moore <sage@neuralmagic.com>
@SageMoore SageMoore requested a review from tlrmchlsmth as a code owner March 11, 2025 17:20
Copy link

👋 Hi! Thank you for contributing to the vLLM project.

💬 Join our developer Slack at https://slack.vllm.ai to discuss your PR in #pr-reviews, coordinate on features in #feat- channels, or join special interest groups in #sig- channels.

Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run fastcheck CI which starts running only a small and essential subset of CI tests to quickly catch errors. You can run other CI tests on top of those by going to your fastcheck build on Buildkite UI (linked in the PR checks section) and unblock them. If you do not have permission to unblock, ping simon-mo or khluu to add you in our Buildkite org.

Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging.

To run CI, PR reviewers can either: Add ready label to the PR or enable auto-merge.

🚀

@mergify mergify bot added the ci/build label Mar 11, 2025
@gshtras
Copy link
Contributor

gshtras commented Mar 11, 2025

Confirming that this fixes the build. Thanks!

Signed-off-by: Sage Moore <sage@neuralmagic.com>
Copy link
Member

@mgoin mgoin left a comment

Choose a reason for hiding this comment

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

Thank you Sage, LGTM. I think this kernels works on SM 70/75 so we should loosing the restriction later cc @jinzhen-lin but let's get the fix in now

@mgoin mgoin added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 11, 2025
Copy link
Contributor

@houseroad houseroad left a comment

Choose a reason for hiding this comment

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

Shall we add a build test to cover such case in the follow up PR?

Ideally, this should be caught by CI directly.

@hongxiayang hongxiayang added the rocm Related to AMD ROCm label Mar 11, 2025
@hongxiayang
Copy link
Collaborator

This is needed to fix ROCm build issue

Copy link
Collaborator

@hongxiayang hongxiayang left a comment

Choose a reason for hiding this comment

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

it fixed my build issue

@hongxiayang
Copy link
Collaborator

Shall we add a build test to cover such case in the follow up PR?

Ideally, this should be caught by CI directly.

I agree.

@SageMoore
Copy link
Contributor Author

Shall we add a build test to cover such case in the follow up PR?

Ideally, this should be caught by CI directly.

Looks like the AMD job was not run on #13321 which is why the build error slipped through the cracks.

Unfortunately, the AMD job has been red for quite some time and we don't want to add additional noise to inflight PRs. We've made decent strides towards getting that build green, but there's more work to do. If you are interested in contributing to this effort, feel free to reach out to me on the vllm slack and I can provide some more context. There's nothing really hard here, we just need to grind through all of the failures.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ci/build ready ONLY add when PR is ready to merge/full CI is needed rocm Related to AMD ROCm
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants