feat(observability): add OpenTelemetry tracing for pipeline parallelism (#23169)

Signed-off-by: Yinzuo Jiang <jiangyinzuo@foxmail.com>
This commit is contained in:
Yinzuo Jiang
2026-04-28 17:05:23 +08:00
committed by GitHub
parent 9a53ab3d6d
commit 71160e4ddb
6 changed files with 68 additions and 8 deletions
+4
View File
@@ -237,6 +237,10 @@ def generate_perfetto_span(engine_root_spans, smg_otel_spans, thread_meta_data):
pid = int(thread_span["attributes"]["pid"])
host_id = thread_span["attributes"]["host_id"]
thread_name = f'{thread_span["attributes"]["host_id"][:8]}:{thread_span["attributes"]["thread_label"]}'
if "pp_rank" in thread_span["attributes"]:
thread_name += f"-PP{thread_span['attributes']['pp_rank']}"
if "dp_rank" in thread_span["attributes"]:
thread_name += f"-DP{thread_span['attributes']['dp_rank']}"
if "tp_rank" in thread_span["attributes"]:
thread_name += f"-TP{thread_span['attributes']['tp_rank']}"