mimo: terse zero_grad_buffer docstrings and add CPU unit test

0c88aab

Select commit

Loading

Failed to load commit list.

Uh oh!

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

DCO

DCO

Auto Swap Labels on: pull_request_target

check-approval

Review Trigger on: pull_request_review

signal

Review Trigger on: pull_request_review

signal

Codeowners Approval Workflow on: push

pre-flight / pre-flight

codeowners-approval

multi-approval-bot-summary

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

Request NVSkills CI on: push

request

Copyright check on: push

pre-flight / pre-flight

copyright-check / main

copyright-check-summary

CICD Megatron-LM on: push 148

is-not-external-contributor

merge-queue-notification

DCO

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

cicd-mbridge-testing-notify

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/engines/test_dynamic_engine.py - latest

tests/unit_tests/inference/test_hybrid_moe.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/test_gated_delta_net.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/a2a_overlap/**/*.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-integration-gate

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_tp2_pp2_dsa - 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 9

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

Review Trigger on: pull_request_review

signal

DCO / DCO succeeded Jun 17, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO

Loading

Re-running checks...