Skip to content

Commit

Permalink
Log process group name instead of pg uid
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#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
  • Loading branch information
shengbao-zheng authored and facebook-github-bot committed Apr 15, 2024
1 parent 7c78556 commit 86c01a5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torch_ucc_tracing.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 86c01a5

Please sign in to comment.