Skip to content

Fix release tests: remove --global-batch-size conflicting with --step-batch-size-schedule - #4545

Merged
ko3n1g merged 1 commit into
NVIDIA:mainfrom
deepakn94:dnarayanan/fix_step_batch_size_schedule_tests
Apr 30, 2026
Merged

Fix release tests: remove --global-batch-size conflicting with --step-batch-size-schedule#4545
ko3n1g merged 1 commit into
NVIDIA:mainfrom
deepakn94:dnarayanan/fix_step_batch_size_schedule_tests

Conversation

@deepakn94

Copy link
Copy Markdown
Contributor

Summary

  • Remove --global-batch-size from 4 GPT 15B release test configs that also specify --step-batch-size-schedule, which are mutually exclusive args
  • Affected tests: gpt3_15b_8t_release, gpt3_15b_8t_release_sm, gpt3_15b_8t_release_gb200, gpt3_15b_8t_release_sm_gb200
  • The --step-batch-size-schedule already defines the batch size ramp (0:384 200B:768 400B:1152), making --global-batch-size redundant

Test plan

  • Verify release tests no longer fail with ValueError: Cannot specify both --step-batch-size-schedule and --global-batch-size

🤖 Generated with Claude Code

…-batch-size-schedule

These two args are mutually exclusive. The step-batch-size-schedule
already defines the batch size ramp, so global-batch-size is redundant.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@copy-pr-bot

copy-pr-bot Bot commented Apr 30, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@deepakn94
deepakn94 marked this pull request as ready for review April 30, 2026 00:14
@svcnvidia-nemo-ci
svcnvidia-nemo-ci requested a review from a team April 30, 2026 00:15
@ko3n1g
ko3n1g added this pull request to the merge queue Apr 30, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown
Contributor

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/25153672872

Merged via the queue into NVIDIA:main with commit 580d53a Apr 30, 2026
69 of 71 checks passed
@deepakn94
deepakn94 deleted the dnarayanan/fix_step_batch_size_schedule_tests branch April 30, 2026 14:03
yhgalaxy pushed a commit to yhgalaxy/Megatron-LM that referenced this pull request Jun 17, 2026
…-batch-size-schedule (NVIDIA#4545)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Signed-off-by: yhgalaxy <yhgalaxy@outlook.com>
jon-barker pushed a commit to jon-barker/Megatron-LM that referenced this pull request Jul 10, 2026
…-batch-size-schedule (NVIDIA#4545)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Signed-off-by: Jon Barker <jbarker@aws-cmh-slurm-1-vscode-02.cm.cluster>
terminator123 pushed a commit to 021ai/Megatron-LM that referenced this pull request Aug 3, 2026
…-batch-size-schedule (NVIDIA#4545)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants