[ci] fix: fix precommit #4950
Workflow file for this run
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
| # # Tests layout | |
| # Each folder under tests/ corresponds to a test category for a sub-namespace in verl. For instance: | |
| # - `tests/trainer` for testing functionality related to `verl/trainer` | |
| # - `tests/models` for testing functionality related to `verl/models` | |
| # - ... | |
| # There are a few folders with `special_` prefix, created for special purposes: | |
| # - `special_distributed`: unit tests that must run with multiple GPUs | |
| # - `special_e2e`: end-to-end tests with training/generation scripts | |
| # - `special_npu`: tests for NPUs | |
| # - `special_sanity`: a suite of quick sanity tests | |
| # - `special_standalone`: a set of test that are designed to run in dedicated environments | |
| # Accelerators for tests | |
| # - By default tests are run with GPU available, except for the ones under `special_npu`, and any test script whose name ends with `on_cpu.py`. | |
| # - For test scripts with `on_cpu.py` name suffix would be tested on CPU resources in linux environment. | |
| # # Workflow layout | |
| # All CI tests are configured by yaml files in `.github/workflows/`. Here's an overview of all test configs: | |
| # 1. A list of always triggered CPU sanity tests: `check-pr-title.yml`, `secrets_scan.yml`, `check-pr-title,yml`, `pre-commit.yml`, `doc.yml` | |
| # 2. Some heavy multi-GPU unit tests, such as `model.yml`, `vllm.yml`, `sgl.yml` | |
| # 3. End-to-end tests: `e2e_*.yml` | |
| # 4. Unit tests | |
| # - `cpu_unit_tests.yml`, run pytest on all scripts with file name pattern `tests/**/test_*_on_cpu.py` | |
| # - `gpu_unit_tests.yml`, run pytest on all scripts with file without the `on_cpu.py` suffix. | |
| # - Since cpu/gpu unit tests by default runs all tests under `tests`, please make sure tests are manually excluded in them when | |
| # - new workflow yaml is added to `.github/workflows` | |
| # - new tests are added to workflow mentioned in 2. | |
| name: e2e_ppo_trainer_megatron_vllm | |
| on: | |
| # Trigger the workflow on push or pull request, | |
| # but only for the main branch. | |
| # For push, for now only anti-patterns are specified so it is more conservative | |
| # and achieves higher coverage. | |
| push: | |
| branches: | |
| - main | |
| - v0.* | |
| paths: | |
| - "**/*.py" | |
| # Other entrypoints | |
| - "!verl/trainer/fsdp_sft_trainer.py" | |
| # Recipes | |
| - "!recipe/**" | |
| # FSDP | |
| - "!verl/workers/**/*dp_*.py" | |
| - "!verl/utils/fsdp_utils.py" | |
| - "!verl/utils/checkpoint/fsdp_checkpoint_manager.py" | |
| - "!verl/model_merger/fsdp_model_merger.py" | |
| pull_request: | |
| branches: | |
| - main | |
| - v0.* | |
| paths: | |
| - "**/*.py" | |
| # Other entrypoints | |
| - "!docker/**" | |
| # Docs | |
| - "!**/*.md" | |
| - "!docs/**" | |
| - "!examples/**" | |
| - "!tests/**" | |
| - "!verl/trainer/main_*.py" | |
| - "!verl/trainer/fsdp_sft_trainer.py" | |
| # Recipes | |
| - "!recipe/**" | |
| # FSDP | |
| - "!verl/workers/**/*dp_*.py" | |
| - "!verl/utils/fsdp_utils.py" | |
| - "!verl/utils/checkpoint/fsdp_checkpoint_manager.py" | |
| - "!verl/model_merger/fsdp_model_merger.py" | |
| # Entrypoints | |
| - ".github/workflows/e2e_ppo_trainer_megatron_vllm.yml" | |
| - "examples/data_preprocess/gsm8k.py" | |
| - "examples/data_preprocess/geo3k.py" | |
| - "tests/special_e2e/run_ppo_trainer_megatron.sh" | |
| - "verl/trainer/main_ppo.py" | |
| - "verl/trainer/config/ppo_megatron_trainer.yaml" | |
| # Cancel jobs on the same ref if a new one is triggered | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref }} | |
| cancel-in-progress: ${{ github.ref != 'refs/heads/main' }} | |
| # Declare permissions just read content. | |
| permissions: | |
| contents: read | |
| env: | |
| IMAGE: "verl-ci-cn-beijing.cr.volces.com/verlai/verl:vllm012.latest" | |
| DYNAMIC_RUNNER_ENDPOINT: "https://sd10g3clalm04ug7alq90.apigateway-cn-beijing.volceapi.com/runner" | |
| TRANSFORMERS_VERSION: "4.56.2" | |
| jobs: | |
| setup: | |
| if: github.repository_owner == 'volcengine' | |
| runs-on: ubuntu-latest | |
| outputs: | |
| runner-label: ${{ steps.create-runner.outputs.runner-label }} | |
| mlp-task-id: ${{ steps.create-runner.outputs.mlp-task-id }} | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - id: create-runner | |
| uses: volcengine/vemlp-github-runner@v1 | |
| with: | |
| mode: "create" | |
| faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}" | |
| mlp-image: "${{ env.IMAGE }}" | |
| # deepseek-ai/deepseek-coder-1.3b-instruct: dense, tie_word_embeddings=False | |
| e2e_ppo_trainer_megatron-deepseek: | |
| needs: setup | |
| runs-on: ["${{ needs.setup.outputs.runner-label || 'L20x8' }}"] | |
| timeout-minutes: 60 # Increase this timeout value as needed | |
| env: | |
| HTTP_PROXY: ${{ secrets.PROXY_HTTP }} | |
| HTTPS_PROXY: ${{ secrets.PROXY_HTTPS }} | |
| NO_PROXY: "localhost,127.0.0.1,hf-mirror.com" | |
| HF_ENDPOINT: "https://hf-mirror.com" | |
| HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable | |
| steps: | |
| - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2 | |
| with: | |
| fetch-depth: 0 | |
| - name: Install the current repository | |
| run: | | |
| pip3 install -r requirements-test.txt | |
| pip3 install --no-deps -e . | |
| pip3 install mbridge | |
| pip3 install math-verify transformers==$TRANSFORMERS_VERSION | |
| - name: Prepare GSM8K dataset | |
| run: | | |
| python3 examples/data_preprocess/gsm8k.py --local_dataset_path ${HOME}/models/hf_data/gsm8k | |
| # Full training save&load | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron, use mbridge e2e to pre-load and save (Deepseek) | |
| run: | | |
| ray stop --force | |
| ALL_OFFLOAD=True SAVE_FREQ=1 MODEL_ID=deepseek-ai/deepseek-coder-1.3b-instruct COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 USE_MBRIDGE=True USE_DIST_CKPT=False \ | |
| bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron, use mbridge e2e to pre-load and save (Deepseek) | |
| run: | | |
| ray stop --force | |
| RESUME_MODE=auto MODEL_ID=deepseek-ai/deepseek-coder-1.3b-instruct TOTAL_TRAIN_STEPS=2 SAVE_FREQ=1 COMMON_PP=4 COMMON_VPP=null COMMON_CP=1 USE_MBRIDGE=True USE_DIST_CKPT=False \ | |
| bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| # LoRA training save&load | |
| - name: clean up and install Megatron-Bridge | |
| run: | | |
| rm -rf checkpoints | |
| pip3 install git+https://github.com/NVIDIA-NeMo/Megatron-Bridge.git@953aabf --no-deps --no-build-isolation | |
| pip3 install git+https://github.com/NVIDIA/Megatron-LM.git@2d398b4 --no-deps --no-build-isolation | |
| pip3 install "nvidia-modelopt[torch]>=0.37.0" transformers==4.57.1 | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron, use Megatron-Bridge LoRA e2e to pre-load and save (Deepseek) | |
| run: | | |
| ray stop --force | |
| ALL_OFFLOAD=True SAVE_FREQ=1 MODEL_ID=deepseek-ai/deepseek-coder-1.3b-instruct COMMON_PP=4 LORA_RANK=8 COMMON_VPP=null COMMON_CP=1 USE_MBRIDGE=True VANILLA_MBRIDGE=False VALUE_VANILLA_MBRIDGE=False USE_DIST_CKPT=False \ | |
| bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron, use Megatron-Bridge LoRA e2e to pre-load and save (Deepseek) | |
| run: | | |
| ray stop --force | |
| RESUME_MODE=auto MODEL_ID=deepseek-ai/deepseek-coder-1.3b-instruct TOTAL_TRAIN_STEPS=2 SAVE_FREQ=1 COMMON_PP=4 LORA_RANK=8 COMMON_VPP=null COMMON_CP=1 USE_MBRIDGE=True VANILLA_MBRIDGE=False VALUE_VANILLA_MBRIDGE=False USE_DIST_CKPT=False \ | |
| bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| - name: clean up | |
| run: | | |
| rm -rf checkpoints | |
| # Qwen3-0.6B: dense, tie_word_embeddings=True | |
| e2e_ppo_trainer_megatron-qwen3: | |
| needs: setup | |
| runs-on: ["${{ needs.setup.outputs.runner-label || 'L20x8' }}"] | |
| timeout-minutes: 60 # Increase this timeout value as needed | |
| env: | |
| HTTP_PROXY: ${{ secrets.PROXY_HTTP }} | |
| HTTPS_PROXY: ${{ secrets.PROXY_HTTPS }} | |
| NO_PROXY: "localhost,127.0.0.1,hf-mirror.com" | |
| HF_ENDPOINT: "https://hf-mirror.com" | |
| HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable | |
| steps: | |
| - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2 | |
| with: | |
| fetch-depth: 0 | |
| - name: Install the current repository | |
| run: | | |
| pip3 install -r requirements-test.txt | |
| pip3 install --no-deps -e . | |
| pip3 install math-verify transformers==$TRANSFORMERS_VERSION | |
| - name: Prepare GSM8K dataset | |
| run: | | |
| python3 examples/data_preprocess/gsm8k.py --local_dataset_path ${HOME}/models/hf_data/gsm8k | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with Megatron (Qwen3) testing learning rate scheduler | |
| run: | | |
| ray stop --force | |
| ALL_OFFLOAD=True VAL_BEFORE_TRAIN=True TEST_FREQ=1 SAVE_FREQ=1 LR_WARMUP_STEPS=1 TOTAL_TRAIN_STEPS=2 MODEL_ID=Qwen/Qwen3-0.6B bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| - name: Running GSM8K E2E training tests with 3D parallelism on 8 L20 GPUs with FP8 rollout | |
| run: | | |
| ray stop --force | |
| export VLLM_USE_V1=1 | |
| ROLLOUT_QUANTIZATION=fp8 TOTAL_TRAIN_STEPS=2 MODEL_ID=Qwen/Qwen3-0.6B bash tests/special_e2e/run_ppo_trainer_megatron.sh | |
| - name: clean up | |
| run: | | |
| rm -rf checkpoints | |
| cleanup: | |
| runs-on: ubuntu-latest | |
| needs: | |
| [ | |
| setup, | |
| e2e_ppo_trainer_megatron-deepseek, | |
| e2e_ppo_trainer_megatron-qwen3, | |
| ] | |
| if: always() | |
| steps: | |
| - id: destroy-runner | |
| uses: volcengine/vemlp-github-runner@v1 | |
| with: | |
| mode: "destroy" | |
| faas-url: "${{ env.DYNAMIC_RUNNER_ENDPOINT }}" | |
| mlp-task-id: "${{ needs.setup.outputs.mlp-task-id }}" |