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

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

Merged

chrisxcai merged 1 commit into ngoyal_changes_for_pp_fp8_jiecaoyu_free_fp16_shard from chriscai_ngoyal_changes_for_pp_fp8_jiecaoyu_free_fp16_shard

May 16, 2024

Merged

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

Only perform cat() during last microbatch backward() within FlattenPa…

Facebook GitHub Tools / Facebook CLA Check succeeded May 15, 2024 in 0s

Contributor License Agreement is valid!

View more details on Facebook GitHub Tools