Fix flush_cache AttributeError on is_stats_logging_rank (#25641)
This commit is contained in:
@@ -3320,7 +3320,7 @@ class Scheduler(
|
|||||||
current_platform.empty_cache()
|
current_platform.empty_cache()
|
||||||
# Per-DP-group leader logs once: ranks within a DP group are
|
# Per-DP-group leader logs once: ranks within a DP group are
|
||||||
# state-synchronous, but DP groups may diverge.
|
# state-synchronous, but DP groups may diverge.
|
||||||
if self.is_stats_logging_rank:
|
if self.metrics_reporter.is_stats_logging_rank:
|
||||||
logger.info("Cache flushed successfully!")
|
logger.info("Cache flushed successfully!")
|
||||||
success = True
|
success = True
|
||||||
else:
|
else:
|
||||||
|
|||||||
Reference in New Issue
Block a user