Skip to content

[megatron-bridge] upgrade to 0.6.2 - #22

Merged
wuxibin89 merged 1 commit into
mainfrom
wuxibin/upgrade
Sep 28, 2026
Merged

wuxibin89 merged 1 commit into
mainfrom
wuxibin/upgrade

Conversation

@wuxibin89

Copy link
Copy Markdown
Contributor

Type of change

  • New image / new build combination (a build_matrix entry, or a new component)
  • Component version bump (ref change in versions.yaml)
  • Feature / enhancement to the pipeline
  • Bug fix
  • Docs only

Summary

Closes #

Dependency changes

What Before After

Validation

  • python3 ci/generate_matrix.py --component <name> | python3 -m json.tool produces the expected matrix
  • bash -n ci/build_scripts/<builder>.sh passes
  • Green workflow_dispatch run:
  • Built wheel installs and imports on the target GPU
  • Release tag/title and wheel_packages assets are as expected
  • Index refreshed and the wheel is pip install-able

New image / new component checklist

  • Submodule added (git submodule add <url> <path>)
  • components: entry in versions.yaml complete (path, ref, builder,
    wheel_packages, torch_cuda_arch_list, requires_cudnn, max_jobs,
    runs_on, extra_env)
  • ci/build_scripts/<builder>.sh added, executable, and leaves wheels in dist/
  • .github/workflows/build-<component>.yml added (copied from
    build-flashinfer.yml for GitHub-hosted, build-apex.yml for self-hosted)
  • README.md component table and repo layout updated
  • No edits needed to build-all.yml / release.yml (both use --component all)

Risk and rollback

@wuxibin89
wuxibin89 merged commit eefaa8d into main Sep 28, 2026
2 checks passed
@github-actions

Copy link
Copy Markdown

Wheelhouse build plan (dry run)

Compared the versions.yaml in this change against the releases on verl-project/verl-wheelhouse. Nothing is built or published from this check - it only shows which matrix rows the next push would rebuild and which it would skip, so a human can confirm the plan matches the intent of the change.

1 row(s) would build, 26 row(s) would be skipped.

Component Ref Arch CUDA Python Torch Decision Reason
apex master x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
apex master x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
apex master aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
apex master aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
deep-ep 3f601f7ac1c062c46502646ff04c535013bfca00 x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
deep-ep 3f601f7ac1c062c46502646ff04c535013bfca00 x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
deep-ep 3f601f7ac1c062c46502646ff04c535013bfca00 aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
deep-ep 3f601f7ac1c062c46502646ff04c535013bfca00 aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
fast-hadamard-transform e7706faf8d1c3b9f241e36860640ad1dac644ede x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
fast-hadamard-transform e7706faf8d1c3b9f241e36860640ad1dac644ede x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
fast-hadamard-transform e7706faf8d1c3b9f241e36860640ad1dac644ede aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
fast-hadamard-transform e7706faf8d1c3b9f241e36860640ad1dac644ede aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-attention v2.8.3 x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-attention v2.8.3 x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-attention v2.8.3 aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-attention v2.8.3 aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-mla 15f13e5030374295491c5ce31b02d7e63a7772c6 x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-mla 15f13e5030374295491c5ce31b02d7e63a7772c6 x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-mla 15f13e5030374295491c5ce31b02d7e63a7772c6 aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flash-mla 15f13e5030374295491c5ce31b02d7e63a7772c6 aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flashinfer v0.6.13 x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
flashinfer v0.6.13 aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
megatron-bridge v0.6.2 x86_64 13.0.2 3.12 2.13.0 🔨 build release not found or unreadable; it would be created
transformer-engine v2.16.1 x86_64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
transformer-engine v2.16.1 x86_64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
transformer-engine v2.16.1 aarch64 13.0.2 3.11 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)
transformer-engine v2.16.1 aarch64 13.0.2 3.12 2.13.0 ⏩ skip exact title, matching build config, and expected wheel packages are present (wheel fatbin re-verification deferred: dry-run mode)

Each row is decided from the release title, the wheelhouse-build-manifest.json release asset (dependency versions, CUDA arch list, extra env, builder command, max_jobs, runs_on, and sha256 fingerprints of the build-toolchain action, the builder script, the lib/ helpers it sources, cuDNN provisioning when required, and declared patches), and wheel package presence. This dry run does not download wheels; the published wheels' fatbin SM-arch coverage is re-verified at push time and gated before upload. Anything that cannot be verified fails closed towards a rebuild.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant