Skip to content

[ROCm][CI] Move ROCm AITER quantization tests - #40938

Merged
tjtanaa merged 10 commits into
vllm-project:mainfrom
ROCm:akaratza_ci_move_rocm_grouped_quant
Aug 18, 2026
Merged

tjtanaa merged 10 commits into
vllm-project:mainfrom
ROCm:akaratza_ci_move_rocm_grouped_quant

Conversation

@AndreasKaratzas

@AndreasKaratzas AndreasKaratzas commented Apr 26, 2026

Copy link
Copy Markdown
Member
  • move ROCm AITER quantization tests into tests/kernels/quantization
  • move MLA FP8 support detection coverage into tests/kernels/attention
  • run the MLA support test in both the MI300 and MI355 Kernels MLA jobs
  • remove the standalone ROCm AITER Ops Test after assigning every test to a maintained kernel suite

The standalone tests/rocm/aiter job duplicated ownership already provided by the kernel test suites. This update rebases the cleanup onto current main, including the AITER hipBLASLt and quantization-schema tests added after the original PR, so removing the standalone job does not leave any tests uncollected. The quantization groups now also depend on vllm/kernels/aiter_ops.py, ensuring changes to the tested RMSNorm registration opt the suite into AMD CI. The MLA support fixture restores its capability cache after every test to avoid leaking fake-module state into adjacent tests.

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@mergify mergify Bot added the rocm Related to AMD ROCm label Apr 26, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Apr 26, 2026

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Code Review

This pull request introduces a new test suite for ROCm AITER grouped FP8 quantization, replacing an older test implementation. The new tests verify the quantization path against a native reference, ensure compatibility with torch.compile, and validate error handling for unsupported configurations. Feedback suggests refining the test skipping logic by using is_aiter_found_and_supported() in the pytestmark to correctly skip tests when the aiter package is missing on supported hardware, rather than triggering assertion failures.

Comment thread tests/kernels/quantization/rocm/aiter/test_grouped_quant.py Outdated
@AndreasKaratzas AndreasKaratzas changed the title [ROCm][CI] Move ROCm grouped quant test [ROCm][CI] Move ROCm AITER quantization tests Apr 26, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@mergify mergify Bot added the ci/build label Apr 26, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@mergify

mergify Bot commented Jul 15, 2026

Copy link
Copy Markdown
Contributor

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @AndreasKaratzas.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>

@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

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #83857 for commit 296a4f75d023.

Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #84275 for commit 2b73ad5ec27f.

@AndreasKaratzas AndreasKaratzas added the ready ONLY add when PR is ready to merge/full CI is needed label Aug 17, 2026
@mergify

mergify Bot commented Aug 17, 2026

Copy link
Copy Markdown
Contributor

Hi @AndreasKaratzas, the pre-commit checks have failed. Please run:

uv pip install pre-commit>=4.5.1
pre-commit install
pre-commit run --all-files

Then, commit the changes and push to your branch.

For future commits, pre-commit will run automatically on changed files before each commit.

@tjtanaa tjtanaa 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.

LGTM

@tjtanaa
tjtanaa merged commit 5fa8ca9 into vllm-project:main Aug 18, 2026
57 of 60 checks passed
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Aug 18, 2026
stefankoncarevic added a commit to ROCm/vllm that referenced this pull request Aug 18, 2026
Resolve the .buildkite/test-amd.yaml conflicts:

- Keep the Quantized Fusions AMD job this branch adds.
- Drop the ROCm AITER Ops Test job. Upstream removed it in vllm-project#40938 when
  tests/rocm/aiter was split into tests/kernels/{attention,quantization},
  so `pytest -v -s rocm/aiter/` would now collect nothing.
- Keep both the streaming_input coverage from this branch and the
  encoder cudagraph coverage from main.

Signed-off-by: Stefan Koncarevic <Stefan.Koncarevic@amd.com>
@AndreasKaratzas
AndreasKaratzas deleted the akaratza_ci_move_rocm_grouped_quant branch August 18, 2026 15:04
wenhuach21 pushed a commit to wenhuach21/vllm that referenced this pull request Aug 19, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Wenhua Cheng <wenhua.cheng@intel.com>
zyp2014 pushed a commit to zyp2014/vllm that referenced this pull request Aug 21, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
wyettzeng pushed a commit to wyettzeng/vllm that referenced this pull request Aug 21, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Wyett <wyettzeng@gmail.com>
zufangzhu pushed a commit to zufangzhu/vllm that referenced this pull request Aug 24, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Zhu, Zufang <zufang.zhu@intel.com>
khushali9 pushed a commit to khushali9/vllm that referenced this pull request Aug 29, 2026
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: khushali9 <khushali.desai9@gmail.com>
eble-amd added a commit to ROCm/vllm that referenced this pull request Sep 10, 2026
Upstream sync 26/N: merge 5fa8ca9 [ROCm][CI] Move ROCm AITER quantization tests (vllm-project#40938) (conflict)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci/build quantization 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.

2 participants