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

Merged

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

JackCaoG merged 1 commit into pytorch:master from zachzzc:defer_to

Don't move full tensor to device in deferred_init

c4721c2f

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG added fsdp

alanwaketan approved these changes on 2023-08-22

JackCaoG merged c57370dc into master 2 years ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

fsdp

Milestone

No milestone