We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9e0b90f commit 29e4a4bCopy full SHA for 29e4a4b
torchrec/distributed/planner/planners.py
@@ -513,10 +513,7 @@ def plan(
513
sharding_plan = to_sharding_plan(best_plan, self._topology)
514
515
end_time = perf_counter()
516
- shall_log_sharding_plan = False
517
for stats in self._stats:
518
- if not isinstance(stats, NoopEmbeddingStats):
519
- shall_log_sharding_plan = True
520
stats.log(
521
sharding_plan=sharding_plan,
522
topology=self._topology,
@@ -531,8 +528,6 @@ def plan(
531
528
sharders=sharders,
532
529
debug=self._debug,
533
530
)
534
- if shall_log_sharding_plan:
535
- logger.info(f"Found sharding plan {sharding_plan}")
536
return sharding_plan
537
else:
538
global_storage_capacity = reduce(
0 commit comments