Skip to content

Separate save_checkpoint into per-type execution paths#3852

Open
Anmol202005 wants to merge 1 commit intoNVIDIA:mainfrom
Anmol202005:checkpoint
Open

Separate save_checkpoint into per-type execution paths#3852
Anmol202005 wants to merge 1 commit intoNVIDIA:mainfrom
Anmol202005:checkpoint

Conversation

@Anmol202005
Copy link

What does this PR do ?

Separate save_checkpoint into per-type helper functions by extracting _save_legacy_checkpoint, _save_local_checkpoint, _save_global_dist_checkpoint, and _save_global_dcp_checkpoint as standalone top-level functions.

Part of #3758.

Pre-checks

  • I have added relevant unit tests
  • I have added relevant functional tests
  • I have added proper typing to my code Typing guidelines
  • I have added relevant documentation
  • I have run the autoformatter.sh on my PR

@copy-pr-bot
Copy link

copy-pr-bot bot commented Mar 13, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@svcnvidia-nemo-ci svcnvidia-nemo-ci marked this pull request as draft March 13, 2026 10:46
@github-actions
Copy link
Contributor

This PR has been automatically converted to draft because all PRs must start as drafts.

When you are ready for review, click Ready for Review to begin the review process. This will:

  1. Add the oncall reviewer (optional reviewer)
  2. Add required review teams based on your changes

See the contribution guide for more details.

@Anmol202005 Anmol202005 marked this pull request as ready for review March 13, 2026 10:49
@svcnvidia-nemo-ci svcnvidia-nemo-ci requested a review from a team March 13, 2026 10:49
@Anmol202005 Anmol202005 marked this pull request as draft March 13, 2026 10:52
@Anmol202005 Anmol202005 marked this pull request as ready for review March 13, 2026 10:54
@chtruong814 chtruong814 added the needs-follow-up Issue needs follow-up label Mar 15, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants