-
Zhenhuan Liu authored
* Fix issues for MCore DDP. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * Remove force data release for CPU offloading. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * Add preserved attributeds. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add main_grad to prevserved attributes. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * Change prepare_for_saving to original tensor and add .data to CPU hook. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update. Signed-off-by:
Dennis Liu <denliu@nvidia.com> * Fix for LayernormLinear in FP8. Signed-off-by:
Dennis Liu <denliu@nvidia.com> --------- Signed-off-by:
Dennis Liu <denliu@nvidia.com> Co-authored-by:
Xin Yao <xiny@nvidia.com> Co-authored-by:
pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
978f1d72