Skip to content

Commit 07b077f

Browse files
test: drop removed runtime_gather_output arg from process_mtp_loss call
MCore b86e75c18 (Fix vocab-parallel MTP loss for TP>1, NVIDIA/Megatron-LM#7682) removed the runtime_gather_output parameter from process_mtp_loss; the Qwen3-VL text model unit test still passed it. Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
1 parent 00d61a5 commit 07b077f

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

‎tests/unit_tests/models/qwen_vl/modelling_qwen3_vl/test_text_model_forward.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,6 @@ def fake_output_layer(hidden, weight=None, runtime_gather_output=None): # noqa:
263263
loss_mask=loss_mask,
264264
output_layer=fake_output_layer,
265265
output_weight=None,
266-
runtime_gather_output=None,
267266
is_training=False,
268267
compute_language_model_loss=fake_language_loss,
269268
config=config,

0 commit comments

Comments
 (0)