PR #8056 Support FSDPv2 compute dtype

Support FSDPv2 compute dtype #8056

lausannel wants to merge 3 commits into pytorch:master from lausannel:master

feat: support spmd distx init and compute dtype

ada8f482

lausannel changed the title ~~Support spmd distx init and compute dtype~~ Support FSDPv2 distx init and compute dtype 1 year ago

fix: add instructions for initializing with `torchdistX`

4d2ee13a

fix: remove torchdistx support

b7977f5f

lausannel changed the title ~~Support FSDPv2 distx init and compute dtype~~ Support FSDPv2 compute dtype 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone