Skip to content

Commit eb9c6d8

Browse files
gnahzgfacebook-github-bot
authored andcommitted
Add comments to min_partition
Summary: As titled Created from CodeHub with https://fburl.com/edit-in-codehub Reviewed By: sarckk Differential Revision: D54489049 fbshipit-source-id: 6365a8051d4d420b18dae702c8769a98ea15f38b
1 parent b572153 commit eb9c6d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrec/distributed/planner/types.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -429,7 +429,7 @@ class ParameterConstraints:
429429

430430
sharding_types: Optional[List[str]] = None
431431
compute_kernels: Optional[List[str]] = None
432-
min_partition: Optional[int] = None # CW sharding
432+
min_partition: Optional[int] = None # CW sharding, min CW dim to shard
433433
pooling_factors: List[float] = field(
434434
default_factory=lambda: [POOLING_FACTOR]
435435
) # average number of embedding lookups required per sample

0 commit comments

Comments
 (0)