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