Skip to content

Commit abde93b

Browse files
committed
Change job name while uploading sharding plan to Manifold
Summary: Adding the planner input context has to the sharding plan name that's uploaded to manifold. We will use this hash in the next diff to validate that the sharding plan loaded from manifold was generated with the same inputs as the model that it's being loaded for. Reviewed By: micrain Differential Revision: D75723272
1 parent 9240bf0 commit abde93b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchrec/distributed/planner/planners.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -496,6 +496,7 @@ def plan(
496496
best_plan=last_proposal,
497497
constraints=self._constraints,
498498
sharders=sharders,
499+
enumerator=self._enumerator,
499500
debug=self._debug,
500501
)
501502

0 commit comments

Comments
 (0)