Skip to content

Commit 0ddad13

Browse files
aliafzalfacebook-github-bot
authored andcommitted
1 parent 61bf3cf commit 0ddad13

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchrec/distributed/embedding_kernel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,9 @@ def create_virtual_table_global_metadata(
105105
# The param size only has the information for my_rank. In order to
106106
# correctly calculate the size for other ranks, we need to use the current
107107
# rank's shard size compared to the shard size of my_rank.
108-
curr_rank_rows = (param.size()[0] * metadata.shards_metadata[rank].shard_sizes[0]) // my_rank_shard_size # pyre-ignore[16]
108+
curr_rank_rows = (
109+
param.size()[0] * metadata.shards_metadata[rank].shard_sizes[0]
110+
) // my_rank_shard_size
109111
else:
110112
curr_rank_rows = (
111113
weight_count_per_rank[rank] if weight_count_per_rank is not None else 1

0 commit comments

Comments
 (0)