Fix GDN DTensor splitting for FSDP checkpointing

df170a2

Select commit

Loading

Failed to load commit list.

Uh oh!

There was an error while loading. Please reload this page.

Sign in for the full log view

Force Draft PR on: pull_request_target

force-draft

Auto Swap Labels on: pull_request_target

check-approval

Claude Complexity Label on: pull_request_target 3

Label PR Complexity

Oncall Assign on: pull_request_target

assign-reviewer

Review Trigger on: pull_request_review

signal

Review Trigger on: pull_request_review

signal

Review Trigger on: pull_request_review

signal

Review Trigger on: pull_request_review

signal

Installation Test on: push

pre-flight / pre-flight

Pip - Python3.12 - AMD64/Linux - NGC PyTorch

UV - Python3.12 - AMD64/Linux - NGC PyTorch

Install test summary

Build docs on: push 1

pre-flight / pre-flight

build-docs / Build docs

build-docs-summary

Copyright check on: push

pre-flight / pre-flight

copyright-check / main

copyright-check-summary

Codeowners Approval Workflow on: push

pre-flight / pre-flight

codeowners-approval

multi-approval-bot-summary

CICD Megatron-LM on: push 134

is-not-external-contributor

merge-queue-notification

pre-flight / pre-flight

cicd-compute-build-matrix

configure

linting

cicd-wait-in-queue

cicd-parse-downstream-testing

cicd-container-build (aws, 766267172432.dkr.ecr.us-east-1.amazonaws.com, nvidia-ci-aws-gpu-x8)

cicd-container-build (gcp, us-east4-docker.pkg.dev/nv-projdgxchipp-20260113193621/megatron-lm, nv...

cicd-mbridge-testing

cicd-parse-unit-tests

tests/unit_tests/test_model_configs.py - latest

tests/unit_tests/test_fp8_param.py - latest

tests/unit_tests/pipeline_parallel/**/*.py - latest

tests/unit_tests/models/**/*.py - latest

tests/unit_tests/data/**/*.py - latest

tests/unit_tests/dist_checkpointing/test_optimizer.py - latest

tests/unit_tests/dist_checkpointing/**/*.py - latest

tests/unit_tests/dist_checkpointing/models/**/*.py - latest

tests/unit_tests/dist_checkpointing/models/test_moe_experts.py - latest

tests/unit_tests/transformer/**/*.py - latest

tests/unit_tests/inference/**/*.py - latest

tests/unit_tests/rl/**/*.py - latest

tests/unit_tests/post_training/**/*.py - latest

tests/unit_tests/resharding/**/*.py - latest

tests/unit_tests/ssm/**/*.py - latest

tests/unit_tests/transformer/moe/**/*.py - latest

tests/unit_tests/distributed/megatron_fsdp/**/*.py - latest

tests/unit_tests/**/*.py - latest

tests/unit_tests/transformer/test_multi_latent_attention.py - latest

tests/unit_tests/transformer/test_multi_token_prediction.py - latest

cicd-parse-integration-tests-h100

cicd-parse-integration-tests-gb200

gpt/gpt3_mcore_te_tp1_pp4_vp1_dist_optimizer_overlap_grad_reduce_param_gather_overlap_optimizer - latest

gpt/gpt3_mcore_te_tp2_pp1_gdn - latest

gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_async - latest

gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_async_mcore - latest

gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_sync - latest

gpt/gpt3_mcore_te_tp4_pp1_resume_torch_dist_dist_optimizer_overlap_grad_reduce_param_gather - latest

moe/gpt3_moe_mcore_te_tp4_ep2_etp2_pp2_resume_torch_dist_dist_optimizer - latest

gpt/gpt3_mcore_te_tp1_pp1_resume_torch_dist_dist_optimizer_1node - latest

gpt/gpt3_mcore_te_tp4_pp1_resume_torch_dist_dist_optimizer_overlap_grad_reduce_param_gather_1node - latest

moe/deepseek_proxy_fsdp_ep2_fsdp2_1node - latest

moe/gpt3_moe_mcore_te_ep8_resume_torch_dist_muon_1node - latest

Nemo_CICD_Test

Coverage_Fake

Coverage (unit-test)

cleanup-taint-node

Build, validate, and release Megatron-Core on: push 8

pre-flight / pre-flight

bump / check-admin-permission

bump / bump-next-version

build-test-publish-wheels / build-and-test-wheels (megatron-core, arm64, quay.io/pypa/manylinux_2_28_aarch64)

build-test-publish-wheels / build-and-test-wheels (megatron-core, amd64, quay.io/pypa/manylinux_2_28_x86_64)

build-test-publish-wheels / build-and-test-wheels (megatron-fsdp, quay.io/pypa/manylinux_2_28_x86_64, amd64)

build-test-publish-wheels / publish-wheels (megatron-core, arm64)

build-test-publish-wheels / publish-wheels (megatron-core, amd64)

build-test-publish-wheels / publish-wheels (megatron-fsdp, amd64)

finalize / create-gh-release

finalize / build-docs / Build docs

finalize / notify

finalize / publish-docs

release-summary

The logs for this run have expired and are no longer available.

Loading

Re-running jobs...