Andrew Gu bd867d691b [FSDP2] Fix backward-compatible imports (#142419) ...

Internal only: the before way meant that `from torch.distributed._composable.fsdp import fully_shard` was importing `fully_shard.py` not the function `fully_shard`. For some reason, the resolution order is different from open source.

To fix this, we match the old import as closely as possible. Namely, we import `fully_shard.py` contents from `.fully_shard`. This should force that import to take precedence.

@diff-train-skip-merge

Differential Revision: [D66990327](https://our.internmc.facebook.com/intern/diff/D66990327)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/142419
Approved by: https://github.com/weifengpy

2024-12-09 23:56:32 +00:00

..

__init__.py

[FSDP2] Fix backward-compatible imports (#142419)

2024-12-09 23:56:32 +00:00

fully_shard.py

[FSDP2] Move to public torch.distributed.fsdp (#141868)

2024-12-07 01:24:28 +00:00