print

fzyzcjy · fzyzcjy · commit fbc96958afe9 · 2025-12-18T21:06:02.000+08:00
diff --git a/python/sglang/srt/managers/scheduler_metrics_mixin.py b/python/sglang/srt/managers/scheduler_metrics_mixin.py
@@ -399,6 +399,7 @@ def log_batch_result_stats(
         if not isinstance(result, GenerationBatchResult):
             return
 
+        print(f"hi log_batch_result_stats {result.expert_distribution_metrics=}")
         if (m := result.expert_distribution_metrics) is not None:
             self.metrics_collector.increment_eplb_balancedness(
                 forward_mode=batch.forward_mode.name.lower(),
diff --git a/python/sglang/srt/model_executor/model_runner.py b/python/sglang/srt/model_executor/model_runner.py
@@ -2749,6 +2749,7 @@ def forward(
                 split_forward_count,
             )
         output.expert_distribution_metrics = recorder_outputs.get("metrics")
+        print(f"hi forward {recorder_outputs=}")
 
         if self.eplb_manager is not None:
             self.eplb_manager.on_forward_pass_end()

Original file line number	Diff line number	Diff line change
`@@ -2749,6 +2749,7 @@ def forward(`
`2749`	`2749`	`split_forward_count,`
`2750`	`2750`	`)`
`2751`	`2751`	`output.expert_distribution_metrics = recorder_outputs.get("metrics")`
	`2752`	`+ print(f"hi forward {recorder_outputs=}")`
`2752`	`2753`
`2753`	`2754`	`if self.eplb_manager is not None:`
`2754`	`2755`	`self.eplb_manager.on_forward_pass_end()`