Skip to content

[FSDP] Don't move full tensor to device in deferred_init#4819

Merged
JackCaoG merged 1 commit intopytorch:masterfrom
zachzzc:defer_to
Aug 22, 2023
Merged

[FSDP] Don't move full tensor to device in deferred_init#4819
JackCaoG merged 1 commit intopytorch:masterfrom
zachzzc:defer_to

Commits

Commits on Mar 25, 2023