Code

Custom (MCORE) FSDP interface#12391

Code

Open

youngeunkwon0405 wants to merge 7 commits intoNVIDIA:mainNVIDIA/NeMo:mainfrom youngeunkwon0405:mcore-fsdp-interfaceyoungeunkwon0405/NeMo:mcore-fsdp-interfaceCopy head branch name to clipboard

+11

Commits

Commits on Feb 26, 2025

• add assertion for the custom fsdp

  Show description for 6aaf614
  youngeunkwon0405

  committed
  6aaf614

  Copy full SHA for 6aaf614

• change it into assertion

  Show description for da27acd
  youngeunkwon0405

  committed
  da27acd

  Copy full SHA for da27acd

• Merge branch 'main' into mcore-fsdp-interface

  youngeunkwon0405

  committed
  806a7a3

  Copy full SHA for 806a7a3

• Apply isort and black reformatting

  Show description for 36dcfa9
  youngeunkwon0405

  committed
  36dcfa9

  Copy full SHA for 36dcfa9

• modify the comments and assertion text

  Show description for 2ceb6c9
  youngeunkwon0405

  committed
  2ceb6c9

  Copy full SHA for 2ceb6c9

• modify the comments

  Show description for ea1cb0e
  youngeunkwon0405

  committed
  ea1cb0e

  Copy full SHA for ea1cb0e

• Add more arg map info

  Show description for eeef832
  youngeunkwon0405

  committed
  eeef832

  Copy full SHA for eeef832