Skip to content

Commit 1d9541b

Browse files
dstaay-fbfacebook-github-bot
authored andcommitted
Fix empty rank broken test (#2746)
Summary: Pull Request resolved: #2746 as per title. Reviewed By: sarckk Differential Revision: D69684770 fbshipit-source-id: 190081b2b91cc05c85f309202691927cf416f352
1 parent a31e14c commit 1d9541b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchrec/distributed/tests/test_model_parallel_hierarchical.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -232,6 +232,7 @@ def test_sharding_nccl_twcw(
232232
def test_sharding_empty_rank(
233233
self, sharding_type: str, variable_batch_per_feature: bool
234234
) -> None:
235+
self._build_tables_and_groups()
235236
table = self.tables[0]
236237
embedding_groups = {"group_0": table.feature_names}
237238
self._run_multi_process_test(

0 commit comments

Comments
 (0)