[CI] Add flash_mla installation script for dsv4 ci tests#24634
Merged
Conversation
Mirrors ci_install_deepep.sh: sources ci_install_dependency.sh to inherit PIP_CMD/venv, skips when flash_mla is already importable (or rebuilds when FORCE_REBUILD_FLASH_MLA=1), symlinks the CUDA 13 cccl include path, and builds with --no-build-isolation. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Move test_deepseek_v4_flash_fp4_b200.py and test_deepseek_v4_flash_fp4_h200.py into their own per-commit suites (stage-c-test-dsv4-4-gpu-b200 and stage-c-test-dsv4-8-gpu-h200) so flash_mla install only runs where needed. The new pr-test.yml jobs invoke ci_install_flash_mla.sh instead of ci_install_dependency.sh and are wired into the wait-for-stage-c gate. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Fridge003
requested review from
Kangyan-Zhou,
bingxche,
ispobock and
merrymercy
as code owners
Contributor
|
Warning You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again! |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
Modifications
Accuracy Tests
Speed Tests and Profiling
Checklist
Review and Merge Process
/tag-and-rerun-ci,/tag-run-ci-label,/rerun-failed-ci