[ROCm][CI] Move ROCm AITER quantization tests - #40938
Conversation
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
There was a problem hiding this comment.
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.
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
|
This pull request has merge conflicts that must be resolved before it can be |
Co-authored-by: OpenAI Codex <codex@openai.com> Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
|
/ci run |
|
✅ Triggered Buildkite CI #83857 for commit |
Co-authored-by: OpenAI Codex <codex@openai.com> Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
|
/ci run |
|
✅ Triggered Buildkite CI #84275 for commit |
|
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-filesThen, commit the changes and push to your branch. For future commits, |
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>
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>
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: 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>
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>
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>
Upstream sync 26/N: merge 5fa8ca9 [ROCm][CI] Move ROCm AITER quantization tests (vllm-project#40938) (conflict)
tests/kernels/quantizationtests/kernels/attentionKernels MLAjobsROCm AITER Ops Testafter assigning every test to a maintained kernel suiteThe standalone
tests/rocm/aiterjob duplicated ownership already provided by the kernel test suites. This update rebases the cleanup onto currentmain, 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 onvllm/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.