feat(observability): add OpenTelemetry tracing for pipeline parallelism (#23169)
Signed-off-by: Yinzuo Jiang <jiangyinzuo@foxmail.com>
This commit is contained in:
@@ -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']}"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user