From 5e0b6776d2db7b2b8ca823e5a443b4b07354a633 Mon Sep 17 00:00:00 2001 From: weli-l <1289113577@qq.com> Date: Thu, 31 Jul 2025 09:40:50 +0800 Subject: [PATCH] use rank to replace local_rank in pytorch Signed-off-by: weli-l <1289113577@qq.com> --- systrace/src/trace/systrace_manager.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/systrace/src/trace/systrace_manager.cc b/systrace/src/trace/systrace_manager.cc index 783d254..0996ee9 100644 --- a/systrace/src/trace/systrace_manager.cc +++ b/systrace/src/trace/systrace_manager.cc @@ -99,7 +99,7 @@ void PyTorchTrace::dumpPyTorchTracing() std::lock_guard lock(trace_mutex_); - pytorch_trace_.set_rank(config::GlobalConfig::Instance().local_rank); + pytorch_trace_.set_rank(config::GlobalConfig::Instance().rank); pytorch_trace_.set_comm(config::GlobalConfig::Instance().job_name); for (size_t i = 0; i < pytorch_tracing_functions_.size(); ++i) @@ -272,4 +272,4 @@ void SysTrace::eventPollerMain() } } -} // namespace systrace \ No newline at end of file +} // namespace systrace -- Gitee