Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP] Make FSDPv1 only perform cat() during last microbatch backward() within FlattenParamsWrapper #1178

Draft
wants to merge 11 commits into
base: ngoyal_changes_for_pp_fp8
Choose a base branch
from

remove .clone()

d95ca8d
Select commit
Loading
Failed to load commit list.
Draft

[WIP] Make FSDPv1 only perform cat() during last microbatch backward() within FlattenParamsWrapper #1178

remove .clone()
d95ca8d
Select commit
Loading
Failed to load commit list.
Facebook GitHub Tools / Facebook CLA Check succeeded Apr 27, 2024 in 0s

Contributor License Agreement is valid!