Super tiny add moe_ep_rank to prometheus labels (#15407)
This commit is contained in:
@@ -78,6 +78,7 @@ class SchedulerMetricsMixin:
|
|||||||
"engine_type": engine_type,
|
"engine_type": engine_type,
|
||||||
"tp_rank": tp_rank,
|
"tp_rank": tp_rank,
|
||||||
"pp_rank": pp_rank,
|
"pp_rank": pp_rank,
|
||||||
|
"moe_ep_rank": self.moe_ep_rank,
|
||||||
}
|
}
|
||||||
if dp_rank is not None:
|
if dp_rank is not None:
|
||||||
labels["dp_rank"] = dp_rank
|
labels["dp_rank"] = dp_rank
|
||||||
|
|||||||
Reference in New Issue
Block a user