Skip to content

[ROCm] Fix and optimize GPT-J-style MRoPE - #49906

Merged
mgoin merged 4 commits into
vllm-project:mainfrom
ROCm:akaratza_mrope_gptj_perf
Jul 28, 2026
Merged

[ROCm] Fix and optimize GPT-J-style MRoPE#49906
mgoin merged 4 commits into
vllm-project:mainfrom
ROCm:akaratza_mrope_gptj_perf

Conversation

@AndreasKaratzas

@AndreasKaratzas AndreasKaratzas commented Jul 26, 2026

Copy link
Copy Markdown
Member
  • Honor the existing adjacent-pair MRoPE setting instead of hardcoding NeoX pairing.
  • Use contiguous ROCm loads and stores with register-only split/interleave, preserving the NeoX and non-ROCm launch paths.

https://buildkite.com/vllm/amd-ci/builds/11254/list?sid=019f966b-1ec6-4bee-8ce6-acf444a98979&tab=output

Assisted-by: OpenAI Codex
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Assisted-by: OpenAI Codex
@mergify mergify Bot added the rocm Related to AMD ROCm label Jul 26, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Jul 26, 2026
@AndreasKaratzas AndreasKaratzas changed the title [ROCm] Support and optimize GPT-J-style MRoPE [ROCm] Fix and optimize GPT-J-style MRoPE Jul 26, 2026
Use the vLLM platform abstraction when selecting the ROCm-specific one-wave launch shape.

Assisted-by: OpenAI Codex
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@AndreasKaratzas
AndreasKaratzas marked this pull request as ready for review July 27, 2026 03:33

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@AndreasKaratzas AndreasKaratzas added the ready ONLY add when PR is ready to merge/full CI is needed label Jul 27, 2026
@tjtanaa

tjtanaa commented Jul 27, 2026

Copy link
Copy Markdown
Member

This PR also changes the behaviour on NVIDIA, it is better to have someone that is able to validate the changes on NVIDIA to make sure there are no change in the performance.

@mgoin mgoin left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Validated on B300

@mgoin
mgoin merged commit 05a0814 into vllm-project:main Jul 28, 2026
100 of 101 checks passed
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Jul 28, 2026
@AndreasKaratzas
AndreasKaratzas deleted the akaratza_mrope_gptj_perf branch July 28, 2026 17:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed rocm Related to AMD ROCm

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants