Skip to content

Delete HeteroEmbeddingShardingPlanner#4169

Closed
micrain wants to merge 1 commit intometa-pytorch:mainfrom
micrain:export-D102251129
Closed

Delete HeteroEmbeddingShardingPlanner#4169
micrain wants to merge 1 commit intometa-pytorch:mainfrom
micrain:export-D102251129

Conversation

@micrain
Copy link
Copy Markdown

@micrain micrain commented Apr 24, 2026

Summary: Remove HeteroEmbeddingShardingPlanner class from planners.py along with its helper function _merge_plans and the unused MemoryBalancedPartitioner import. Also remove all references in sharding_config.py (import, "hetero" planner branch, docstring), delete the test_infer_hetero_shardings.py test file, and clean up corresponding BUCK targets.

Reviewed By: aporialiao

Differential Revision: D102251129

Summary: Remove HeteroEmbeddingShardingPlanner class from planners.py along with its helper function _merge_plans and the unused MemoryBalancedPartitioner import. Also remove all references in sharding_config.py (import, "hetero" planner branch, docstring), delete the test_infer_hetero_shardings.py test file, and clean up corresponding BUCK targets.

Reviewed By: aporialiao

Differential Revision: D102251129
@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 24, 2026
@meta-codesync
Copy link
Copy Markdown
Contributor

meta-codesync Bot commented Apr 24, 2026

@micrain has exported this pull request. If you are a Meta employee, you can view the originating Diff in D102251129.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant