Skip to content

Commit aee1fa6

Browse files
committed
Change job name while uploading sharding plan to Manifold (#3092)
Summary: Pull Request resolved: #3092 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 1885423 commit aee1fa6

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)