Add IsHeadNode tag to node metrics

Signed-off-by: Vignesh Hirudayakanth <vignesh@anyscale.com>
ray-project · Oct 24, 2024 · a7bd86b · a7bd86b
1 parent 289374a
commit a7bd86b
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 52 deletions.
diff --git a/python/ray/dashboard/consts.py b/python/ray/dashboard/consts.py
@@ -65,7 +65,12 @@
 RAY_JOB_START_TIMEOUT_SECONDS_ENV_VAR = "RAY_JOB_START_TIMEOUT_SECONDS"
 # Port that dashboard prometheus metrics will be exported to
 DASHBOARD_METRIC_PORT = env_integer("DASHBOARD_METRIC_PORT", 44227)
+
+NODE_TAG_KEYS = ["ip", "Version", "SessionName", "IsHeadNode"]
+GPU_TAG_KEYS = NODE_TAG_KEYS + ["GpuDeviceName", "GpuIndex"]
+CLUSTER_TAG_KEYS = ["node_type", "Version", "SessionName"]
 COMPONENT_METRICS_TAG_KEYS = ["ip", "pid", "Version", "Component", "SessionName"]
+
 # Dashboard metrics are tracked separately at the dashboard. TODO(sang): Support GCS.
 AVAILABLE_COMPONENT_NAMES_FOR_METRICS = {
     "workers",