Skip to content

Commit

Permalink
Remove import HeteroPlanner to fix torch package issues
Browse files Browse the repository at this point in the history
Differential Revision: D56358725
  • Loading branch information
Ivan Kobzarev authored and facebook-github-bot committed Apr 19, 2024
1 parent 254d60d commit 12105af
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions torchrec/distributed/planner/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,6 @@
- automatically building and selecting an optimized sharding plan.
"""

from torchrec.distributed.planner.planners import ( # noqa
EmbeddingShardingPlanner,
HeteroEmbeddingShardingPlanner, # noqa
)
from torchrec.distributed.planner.planners import EmbeddingShardingPlanner # noqa
from torchrec.distributed.planner.types import ParameterConstraints, Topology # noqa
from torchrec.distributed.planner.utils import bytes_to_gb, sharder_name # noqa

0 comments on commit 12105af

Please sign in to comment.