-
-
Notifications
You must be signed in to change notification settings - Fork 10.7k
[CI Bugfix] Fix CI OOM for test_shared_storage_connector_hashes
#21973
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
[CI Bugfix] Fix CI OOM for test_shared_storage_connector_hashes
#21973
Conversation
…connector_hashes Signed-off-by: mgoin <[email protected]>
👋 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 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 🚀 |
test_shared_storage_connector_hashes
test_shared_storage_connector_hashes
There was a problem hiding this comment.
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 aims to fix a recurring out-of-memory (OOM) error in the test_shared_storage_connector_hashes
CI test. The changes introduce gpu_memory_utilization=0.4
and enforce_eager=True
to the EngineArgs
for this specific test. This is a sound approach to reduce memory consumption in a resource-constrained CI environment, as it limits the KV cache size and disables CUDA graphs which can be memory-intensive. The changes are localized to the test and appear correct.
Signed-off-by: mgoin <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for fixing
…lm-project#21973) Signed-off-by: mgoin <[email protected]>
from vllm.multimodal.utils import encode_image_base64 | ||
|
||
MODEL_NAME = "Qwen/Qwen2.5-VL-3B-Instruct" | ||
MODEL_NAME = "RedHatAI/Qwen2.5-VL-3B-Instruct-quantized.w4a16" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks like this model also breaks the CI unfortunately:
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] ValueError: Failed to find a kernel that can implement the WNA16 linear layer. Reasons:
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] MacheteLinearKernel requires capability 90, current compute capability is 89
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] AllSparkLinearKernel cannot implement due to: For Ampere GPU, AllSpark does not support group_size = 128. Only group_size = -1 are supported.
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] MarlinLinearKernel cannot implement due to: Weight output_size_per_partition = 6840 is not divisible by min_thread_n = 64. Consider reducing tensor_parallel_size or running with --quantization gptq.
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] Dynamic4bitLinearKernel cannot implement due to: Only CPU is supported
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] BitBLASLinearKernel cannot implement due to: bitblas is not installed. Please install bitblas by running `pip install bitblas>=0.1.0`
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] ConchLinearKernel cannot implement due to: conch-triton-kernels is not installed, please install it via `pip install conch-triton-kernels` and try again!
[2025-08-03T23:44:26Z] (EngineCore_0 pid=14405) ERROR 08-03 16:44:26 [core.py:683] ExllamaLinearKernel cannot implement due to: Exllama only supports float16 activations
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Did we change the device running this test?
…lm-project#21973) Signed-off-by: mgoin <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]> Signed-off-by: x22x22 <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]> Signed-off-by: Jinzhen Lin <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]> Signed-off-by: Noam Gat <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]> Signed-off-by: Paul Pak <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]> Signed-off-by: Diego-Castan <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]>
…lm-project#21973) Signed-off-by: mgoin <[email protected]>
Purpose
This test has been consistently failing since its addition in #21611
https://buildkite.com/vllm/ci/builds/25475/steps/canvas?sid=01985c5a-a224-435b-ab54-41cb1837705a#01985c5a-a380-4372-bf58-7de2780ad329/7-13621
History https://buildkite.com/organizations/vllm/analytics/suites/ci-1/tests/7b01abb7-8064-8f64-9362-7eceb5b9280e?period=7days

Test Plan
Green CI
Test Result