Repository navigation
Commit 1e7d176
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&size=40){kind=avatar}
chore(beep boop 🤖): Bump
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>uv.lock (r0.4.1, mcore-core_r0.17.0) (2026-05-21)1 parent aeefb3a commit 1e7d176
3 files changed
Lines changed: 742 additions & 994 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
Submodule Megatron-LM updated 57 files
- docker/Dockerfile.ci.dev+1-1
- docs/conf.py+2-2
- docs/project.json+1-1
- docs/versions1.json+7-2
- megatron/core/__init__.py+7-1
- megatron/core/dist_checkpointing/serialization.py+30-1
- megatron/core/dist_checkpointing/strategies/common.py+2-2
- megatron/core/dist_checkpointing/strategies/fully_parallel.py+1-1
- megatron/core/dist_checkpointing/strategies/nvrx.py+87
- megatron/core/dist_checkpointing/strategies/torch.py+27-23
- megatron/core/dist_checkpointing/validation.py+154-1
- megatron/core/distributed/distributed_data_parallel.py+4-3
- megatron/core/distributed/distributed_data_parallel_config.py+4
- megatron/core/distributed/fsdp/src/megatron_fsdp/package_info.py+2-2
- megatron/core/distributed/param_and_grad_buffer.py+168-20
- megatron/core/extensions/transformer_engine.py+3-1
- megatron/core/fp4_utils.py+81
- megatron/core/inference/inference_request.py+22-31
- megatron/core/inference/text_generation_server/dynamic_text_gen_server/endpoints/chat_completions.py+45-12
- megatron/core/optimizer/distrib_optimizer.py+64-7
- megatron/core/package_info.py+2-2
- megatron/core/safe_globals.py+56
- megatron/core/transformer/moe/fused_a2a.py+91-19
- megatron/core/transformer/moe/token_dispatcher.py+5-1
- megatron/core/transformer/transformer_config.py+16-3
- megatron/training/arguments.py+19-6
- megatron/training/async_utils.py+33-10
- megatron/training/checkpointing.py+14-12
- megatron/training/config/training_config.py+19-5
- megatron/training/inprocess_restart.py+10-7
- megatron/training/training.py+1-7
- megatron/training/utils.py+7-1
- pyproject.toml+5-6
- tests/functional_tests/shell_test_utils/run_ci_test.sh+12-1
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_async/model_config.yaml+87
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_async_mcore/model_config.yaml+87
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp1_gdn_no_nvrx_sync/model_config.yaml+84
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp2_resume_torch_dist/model_config.yaml+1
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp2_resume_torch_dist_cp2_nondeterministic/model_config.yaml+1
- tests/functional_tests/test_cases/gpt/gpt3_mcore_te_tp2_pp2_resume_torch_dist_reshard_1x4xNone/model_config.yaml+1
- tests/functional_tests/test_cases/gpt/gpt3_mcore_tp1_pp1_resume_torch_dist_dist_optimizer_overlap_grad_reduce_param_gather/model_config.yaml+1
- tests/functional_tests/test_cases/moe/gpt3_mcore_te_tp1_pp2_resume_torch_dist_reshard_2x1x4_te_8experts2parallel_dist_optimizer/model_config.yaml+1
- tests/functional_tests/test_cases/moe/gpt3_mcore_te_tp2_pp1_resume_torch_dist_te_8experts2parallel_dist_optimizer/model_config.yaml+1
- tests/test_utils/recipes/h100/gpt.yaml+15
- tests/unit_tests/conftest.py+5-1
- tests/unit_tests/dist_checkpointing/test_async_save.py+56
- tests/unit_tests/dist_checkpointing/test_integrity.py+106
- tests/unit_tests/dist_checkpointing/test_pipeline_parallel_layout.py+1
- tests/unit_tests/dist_checkpointing/test_safe_globals.py+16-1
- tests/unit_tests/inference/contexts/test_dynamic_prefix_caching.py+42-2
- tests/unit_tests/models/test_mamba_moe_model.py+4-1
- tests/unit_tests/pipeline_parallel/test_pipeline_layout.py+1
- tests/unit_tests/test_checkpointing.py+1
- tests/unit_tests/test_fp4_param.py+349
- tests/unit_tests/test_utilities.py+9-1
- tests/unit_tests/transformer/moe/test_token_dispatcher.py+41-3
- uv.lock+1.4k-2.4k
0 commit comments