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.

Sign up for GitHub

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

Jump to bottom

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #1180

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

use grad instead of grad.data

b5e138f
Select commit
Loading
Failed to load commit list.
Draft

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #1180

use grad instead of grad.data
b5e138f
Select commit
Loading
Failed to load commit list.
Facebook GitHub Tools / Facebook CLA Check succeeded May 15, 2024 in 0s

Contributor License Agreement is valid!

View more details on Facebook GitHub Tools