From 86c01a5c89fe9ce22cec4a9d6bef5dd61c9ee74a Mon Sep 17 00:00:00 2001
From: Shengbao Zheng <shengbao@meta.com>
Date: Mon, 15 Apr 2024 14:40:40 -0700
Subject: [PATCH] Log process group name instead of pg uid
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Summary:
X-link: https://github.com/pytorch/pytorch/pull/124035

As part of the work of unifying process group identifier, log <group_name, group_desc>,  instead of pg uid in profiler.
- group_name remains as the unique identifier, e.g. “0”, "1"
- group_desc will be the user specified name, e.g. "fsdp".

Reviewed By: aaronenyeshi, kwen2501

Differential Revision: D55610682

fbshipit-source-id: 01c9ad38a7ac06f4c40156ae6e9e20cdda989ca7
---
 src/torch_ucc_tracing.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torch_ucc_tracing.cpp b/src/torch_ucc_tracing.cpp
index 1106b42..8ac904e 100644
--- a/src/torch_ucc_tracing.cpp
+++ b/src/torch_ucc_tracing.cpp
@@ -169,7 +169,7 @@ void CommTraceLogger::recordComms(
   // record the trace to kineto trace if applicable
   RECORD_PARAM_COMMS(
       static_cast<int64_t>(seqnum), // seq
-      0, // process group ptr
+      std::make_tuple("0", ""), // pg_name tuple
       rank,
       commName.c_str(),
       inSize,