Repository navigation
fix(ci): fix H200 nightly benchmark model path, worker logs and CUDA errors - #411
Conversation
📝 WalkthroughWalkthroughThis PR enables nightly E2E mode in CI (env + log dir), mounts local model paths into test Docker containers, removes vLLM's --enforce-eager flag, and conditionally adjusts model vllm_args and worker_args for nightly runs. Changes
Estimated code review effort🎯 3 (Moderate) | ⏱️ ~20 minutes Possibly related PRs
Suggested reviewers
Poem
🚥 Pre-merge checks | ✅ 4✅ Passed checks (4 passed)
✏️ Tip: You can configure your own custom pre-merge checks in the settings. ✨ Finishing touches
🧪 Generate unit tests (beta)
No actionable comments were generated in the recent review. 🎉 Comment |
Summary of ChangesHello @key4ng, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed! This pull request resolves an issue related to model path handling, specifically for H200 models, within the benchmark testing setup. It enhances the command builder to correctly mount local model directories and streamlines the process of passing necessary environment variables to the container, ensuring models are accessible for testing. Highlights
Changelog
Using Gemini Code AssistThe full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips. Invoking Gemini You can request assistance from Gemini at any point by creating a comment using either
Customization To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a Limitations & Feedback Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here. You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension. Footnotes
|
There was a problem hiding this comment.
Code Review
This pull request introduces a mechanism to mount a local model directory into the benchmark container, which is useful for environments like H200 where models are stored locally. The implementation correctly uses the ROUTER_LOCAL_MODEL_PATH environment variable to configure the volume mount. However, the change also removes the pass-through of the HF_HOME environment variable, which could be an unintended breaking change. My review includes a suggestion to restore this functionality to avoid breaking existing workflows.
|
gpt-oss grpc vllm test time reduced to 1 hr |
There was a problem hiding this comment.
Actionable comments posted: 2
🤖 Fix all issues with AI agents
In `@e2e_test/infra/model_specs.py`:
- Around line 105-106: Update the stale inline comment next to the
"--mem-fraction-static=0.82" argument in the list (near the
"--attention-backend=fa3" entry): change the comment text from "90% GPU memory
for static allocation" to reflect the actual value, e.g. "82% GPU memory for
static allocation", so the comment matches the "--mem-fraction-static=0.82"
setting.
- Line 111: The inline comment for the argument "--attention-backend=FLASHINFER"
is misleading (it says "fa3 attention backend"); update the comment to correctly
indicate that this flag selects the FLASHINFER attention backend (vLLM
equivalent) rather than fa3, so change the comment next to
"--attention-backend=FLASHINFER" in model_specs.py to reference FLASHINFER/vLLM.
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
761f18c to
ca89bb1
Compare
…errors (#411) Signed-off-by: ppraneth <pranethparuchuri@gmail.com>
…errors (#411) Signed-off-by: ppraneth <pranethparuchuri@gmail.com>
Description
Problem
The H200 nightly benchmark tests had three issues:
E2E_LOG_DIRwas not set.cudaErrorIllegalAddress(illegal memory access) due to aggressive memory settings and incompatible FlashInfer MLA attention backend configuration.Solution
E2E_LOG_DIRandmkdir -pto the H200 single-worker and H100 multi-worker jobs so worker logs are written to files and included in uploaded artifacts.flashinfer(with MLA-specific flags) tofa3, and reduce--mem-fraction-staticfrom 0.9 to 0.82 to avoid out-of-bounds memory access.Changes
.github/workflows/nightly-benchmark.yml: AddE2E_LOG_DIR: nightly_gateway_logsandmkdir -p nightly_gateway_logstomulti-workerandsingle-worker-h200jobse2e_test/infra/model_specs.py: Replace FlashInfer MLA backend flags with--attention-backend=fa3and lower static memory fraction to 0.82 for Llama-4-Mavericke2e_test/benchmarks/conftest.py: MountROUTER_LOCAL_MODEL_PATHinto benchmark container.github/workflows/nightly-benchmark.yml: Temporary concurrency group renameTest Plan
Checklist
cargo +nightly fmtpassescargo clippy --all-targets --all-features -- -D warningspassesSummary by CodeRabbit
New Features
Chores