Repository navigation
Commit aba6ed2
File tree
- 3rdparty
- docs
- _static/model-verification
- fern
- versions
- nightly/pages/models
- muse_spark_open
- nemotron
- models
- muse_spark_open
- nemotron
- examples
- model_verification_cards
- k-exaone-2
- muse-spark-open-816b-a42b
- models/muse_spark_open
- scripts
- docs
- training
- src/megatron/bridge
- data/collators
- models
- conversion
- hf_pretrained
- muse_glimmer
- muse_spark_open
- modeling_muse_spark_open
- recipes
- muse_spark_open
- gb200
- tests
- functional_tests
- launch_scripts
- gb200/active
- h100/active
- test_groups/models/muse_spark_open
- unit_tests
- doc_consistency
- models
- hf_pretrained
- muse_spark_open
- recipes
- scripts/training
- skills/create_model_verification_card
Some content is hidden
Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
| 1 | + | |
Submodule Megatron-LM updated 97 files
- .github/copy-pr-bot.yaml+1-1
- .github/workflows/_build_ci_container.yml+2-1
- docs/user-guide/features/cuda_graph.md+5-7
- examples/inference/README.md+4-4
- examples/inference/advanced/gpt_dynamic_inference_with_coordinator.py+12
- examples/mimo/model_providers/llava_avlm.py+5-2
- examples/mimo/model_providers/llava_vlm.py+4-1
- examples/mimo/model_providers/mock.py+4-1
- examples/mimo/model_providers/nemotron_moe_vlm.py+6-1
- examples/mimo/model_providers/radio_encoder.py+1
- examples/mimo/train.py+25-2
- examples/mimo/training/args.py+17-6
- examples/mimo/training/builder.py+1
- examples/mimo/training/data.py+4-11
- examples/mimo/training/topology.py+77-34
- examples/mimo/utils/hetero.py+26
- megatron/core/context_parallel/utils.py+35-9
- megatron/core/distributed/fsdp/mcore_fsdp_adapter.py+2-6
- megatron/core/distributed/fsdp/src/megatron_fsdp/experimental/module.py+18
- megatron/core/full_cuda_graph.py+14-5
- megatron/core/inference/contexts/dynamic_context.py+5-1
- megatron/core/inference/contexts/kv_block_allocator.py+22-24
- megatron/core/inference/contexts/routing_metadata.py+44-18
- megatron/core/inference/engines/dynamic_engine.py+181-33
- megatron/core/inference/moe/batch_invariant.py+6-4
- megatron/core/inference/text_generation_controllers/text_generation_controller.py+574-159
- megatron/core/models/mimo/model/base.py+96-162
- megatron/core/models/mimo/partition/utils.py+112-161
- megatron/core/pipeline_parallel/bridge_communicator.py+16-11
- megatron/core/rerun_state_machine.py+51-1
- megatron/core/ssm/gated_delta_net/gdn.py+18-28
- megatron/core/ssm/ops/gdp/__init__.py+2-1
- megatron/core/ssm/ops/gdp/common.py+32-3
- megatron/core/ssm/ops/gdp/fused_recurrent.py+46-8
- megatron/core/ssm/ssm_inference.py+9
- megatron/core/transformer/custom_layers/batch_invariant_kernels.py+6-5
- megatron/core/transformer/experimental_attention_variant/absorbed_mla.py+162-113
- megatron/core/transformer/experimental_attention_variant/dsa.py+824-106
- megatron/core/transformer/experimental_attention_variant/dsa_cudnn_kernels.py+3
- megatron/core/transformer/experimental_attention_variant/ops/tilelang_dsa.py+6-2
- megatron/core/transformer/moe/router.py+1-1
- megatron/core/transformer/moe/router_replay.py+115-27
- megatron/core/transformer/multi_token_prediction.py+5-4
- megatron/core/transformer/transformer_config.py+85-7
- megatron/core/utils.py+14-7
- megatron/training/arguments.py+10-6
- tests/functional_tests/dynamic_inference_functional_tests.md+1-1
- tests/functional_tests/test_cases/moe/gpt_dynamic_inference_tp4_etp1_pp1_ep8_16B_routing_replay_async_sched/golden_values_dev_dgx_h100.json+1
- tests/functional_tests/test_cases/moe/gpt_dynamic_inference_tp4_etp1_pp1_ep8_16B_routing_replay_async_sched/model_config.yaml+85
- tests/functional_tests/test_cases/moe/gpt_dynamic_inference_tp4_etp1_pp1_ep8_16B_routing_replay_prefix_cache_cudagraph_async_sched/golden_values_dev_dgx_h100.json+2.3k
- tests/functional_tests/test_cases/moe/gpt_dynamic_inference_tp4_etp1_pp1_ep8_16B_routing_replay_prefix_cache_cudagraph_async_sched/model_config.yaml+91
- tests/functional_tests/test_cases/moe/gpt_dynamic_inference_tp4_etp1_pp1_ep8_16B_routing_replay_prefix_cache_cudagraph_async_sched/prompts.jsonl+2
- tests/test_utils/recipes/h100/moe-dynamic-inference-with-coordinator.yaml+11
- tests/unit_tests/data/test_get_batch.py+38-5
- tests/unit_tests/determinism/kernels/manifest.py+12-1
- tests/unit_tests/determinism/kernels/test_fused_triton_kernels.py+59-1
- tests/unit_tests/determinism/kernels/test_inference_kernels.py+84
- tests/unit_tests/determinism/kernels/test_ssm_kernels.py+13-1
- tests/unit_tests/determinism/kernels/test_te_wrappers.py+44
- tests/unit_tests/distributed/mfsdp_v2/test_fully_shard.py+79
- tests/unit_tests/distributed/mfsdp_v2/test_mcore_adapter.py+99-89
- tests/unit_tests/inference/contexts/test_dynamic_prefix_caching.py+33-7
- tests/unit_tests/inference/contexts/test_kv_event_publication.py+5-5
- tests/unit_tests/inference/contexts/test_routing_metadata.py+128-4
- tests/unit_tests/inference/engines/test_dynamic_engine.py+11-8
- tests/unit_tests/inference/engines/test_dynamic_engine_async_sched.py+36-28
- tests/unit_tests/inference/engines/test_dynamic_engine_parallel_and_chunked_prefill.py+15-3
- tests/unit_tests/inference/engines/test_request_finalization.py+176
- tests/unit_tests/inference/engines/test_shortcut_moe_cuda_graph_e2e.py+105
- tests/unit_tests/inference/test_mimo_text_checkpoint_detection.py+29
- tests/unit_tests/inference/test_moe_dispatching_and_routing.py+66-5
- tests/unit_tests/inference/test_wandb_logging.py+2-2
- tests/unit_tests/inference/text_generation_controllers/test_text_generation_controller.py+319-45
- tests/unit_tests/models/mimo/test_mimo_1f1b_schedule.py+4-2
- tests/unit_tests/models/mimo/test_mimo_embedding_alignment.py+2-3
- tests/unit_tests/models/mimo/test_mimo_gtp_cp_layout.py+185
- tests/unit_tests/models/mimo/test_mimo_hetero_grid_args.py+47
- tests/unit_tests/models/mimo/test_mimo_mock_data.py+45-4
- tests/unit_tests/models/mimo/test_mimo_model.py+72-39
- tests/unit_tests/models/mimo/test_mimo_partition.py+159-130
- tests/unit_tests/models/mimo/test_nemotron_moe_vlm_provider.py+47-2
- tests/unit_tests/models/test_hybrid_moe_model.py+4
- tests/unit_tests/pipeline_parallel/test_bridge_communicator.py+38-4
- tests/unit_tests/ssm/test_gated_delta_net_inference.py+15-10
- tests/unit_tests/ssm/test_gdp_dynamic_inference.py+39-13
- tests/unit_tests/test_utilities.py+7
- tests/unit_tests/transformer/experimental_attention_variant/test_attention_variant_dsa.py+133
- tests/unit_tests/transformer/experimental_attention_variant/test_dsa_tilelang_kernels.py+31-3
- tests/unit_tests/transformer/experimental_attention_variant/test_kpool_causal_tail.py+328
- tests/unit_tests/transformer/moe/test_aux_loss.py+11-9
- tests/unit_tests/transformer/moe/test_router_replay.py+32
- tests/unit_tests/transformer/test_full_cuda_graph_validation.py+147
- tests/unit_tests/transformer/test_mtp_graph_logging.py+69
- tests/unit_tests/transformer/test_multi_token_prediction.py+6
- tests/unit_tests/transformer/test_residual_recompute.py+1-1
- tests/unit_tests/transformer/test_transformer_config.py+45-2
- tests/unit_tests/transformer/test_wide_residual_connection.py+5-1
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
11 | 11 | | |
12 | 12 | | |
13 | 13 | | |
| 14 | + | |
| 15 | + | |
14 | 16 | | |
15 | 17 | | |
16 | 18 | | |
| |||
242 | 244 | | |
243 | 245 | | |
244 | 246 | | |
| 247 | + | |
245 | 248 | | |
246 | 249 | | |
247 | 250 | | |
| |||
Large diffs are not rendered by default.
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
57 | 57 | | |
58 | 58 | | |
59 | 59 | | |
60 | | - | |
61 | | - | |
62 | | - | |
63 | 60 | | |
64 | 61 | | |
65 | 62 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
21 | 21 | | |
22 | 22 | | |
23 | 23 | | |
24 | | - | |
25 | | - | |
26 | | - | |
27 | | - | |
28 | | - | |
29 | | - | |
30 | | - | |
31 | | - | |
32 | | - | |
33 | | - | |
34 | | - | |
35 | | - | |
36 | | - | |
37 | | - | |
38 | | - | |
39 | | - | |
40 | | - | |
41 | | - | |
42 | | - | |
43 | | - | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
124 | 124 | | |
125 | 125 | | |
126 | 126 | | |
127 | | - | |
| 127 | + | |
| 128 | + | |
| 129 | + | |
| 130 | + | |
128 | 131 | | |
129 | 132 | | |
130 | 133 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
4 | 4 | | |
5 | 5 | | |
6 | 6 | | |
7 | | - | |
| 7 | + | |
8 | 8 | | |
9 | 9 | | |
10 | 10 | | |
| |||
72 | 72 | | |
73 | 73 | | |
74 | 74 | | |
75 | | - | |
| 75 | + | |
76 | 76 | | |
77 | 77 | | |
78 | 78 | | |
79 | 79 | | |
80 | 80 | | |
81 | 81 | | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
82 | 86 | | |
83 | 87 | | |
84 | 88 | | |
| |||
112 | 116 | | |
113 | 117 | | |
114 | 118 | | |
115 | | - | |
| 119 | + | |
116 | 120 | | |
117 | 121 | | |
118 | 122 | | |
| |||
141 | 145 | | |
142 | 146 | | |
143 | 147 | | |
| 148 | + | |
| 149 | + | |
| 150 | + | |
| 151 | + | |
| 152 | + | |
| 153 | + | |
| 154 | + | |
| 155 | + | |
144 | 156 | | |
145 | 157 | | |
146 | 158 | | |
| |||
0 commit comments