From e96876948105110b6542d1e8e1db32df58afcd02 Mon Sep 17 00:00:00 2001 From: Marvin Dsouza Date: Mon, 16 Mar 2026 11:05:23 -0700 Subject: [PATCH] Add NCCL sequence number (seq_num) field mapping to HTA event args config (#322) Summary: Pull Request resolved: https://github.com/facebookresearch/HolisticTraceAnalysis/pull/322 Add the nccl::seq field mapping so HTA can parse the Seq field from Kineto chrome trace event args into a seq_num column. This enables cross-rank correlation of collective operations in HTA analysis. Changes: - default_event_args.py: Add nccl::seq mapping (raw_name: Seq -> seq_num) - event_args_1.0.0.yaml: Same mapping for versioned config Differential Revision: D96145502 --- hta/configs/default_event_args.py | 5 +++++ hta/configs/event_args_formats/event_args_1.0.0.yaml | 5 +++++ 2 files changed, 10 insertions(+) diff --git a/hta/configs/default_event_args.py b/hta/configs/default_event_args.py index ec3afdb..ee51b8a 100644 --- a/hta/configs/default_event_args.py +++ b/hta/configs/default_event_args.py @@ -250,6 +250,11 @@ raw_name: Rank value_type: Int default_value: -1 + nccl::seq: + name: seq_num + raw_name: Seq + value_type: Int + default_value: -1 inference::batch_size: name: batch_size raw_name: batchSize diff --git a/hta/configs/event_args_formats/event_args_1.0.0.yaml b/hta/configs/event_args_formats/event_args_1.0.0.yaml index 0497df5..bdc2c76 100644 --- a/hta/configs/event_args_formats/event_args_1.0.0.yaml +++ b/hta/configs/event_args_formats/event_args_1.0.0.yaml @@ -249,6 +249,11 @@ AVAILABLE_ARGS: raw_name: Rank value_type: Int default_value: -1 + nccl::seq: + name: seq_num + raw_name: Seq + value_type: Int + default_value: -1 inference::batch_size: name: batch_size raw_name: batchSize