Skip to content

Commit 8a2105e

Browse files
authored
Add missed log for reduce_scatter_tensor_coalesced (#1808)
1 parent 2d80827 commit 8a2105e

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

src/xccl/ProcessGroupXCCL.cpp

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1740,6 +1740,26 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::reduce_scatter_tensor_coalesced(
17401740
std::vector<at::Tensor>& outputs,
17411741
std::vector<at::Tensor>& inputs,
17421742
const ReduceScatterOptions& opts) {
1743+
1744+
RECORD_PARAM_COMMS_DATA_WITH_LOG(
1745+
std::make_tuple(
1746+
static_cast<int64_t>(seqCollective_) + 1,
1747+
false), // seq + 1 to match collective and assume only one collective
1748+
// in coalesced range
1749+
std::make_tuple(pg_uid_, pg_desc_), // PG name tuple
1750+
inputs, // inputTensors
1751+
outputs, // outputTensors
1752+
rank_, // rank
1753+
"reduce_scatter_tensor_coalesced", // collective name
1754+
getTensorsNumel(inputs), // inNelems
1755+
getTensorsNumel(outputs), // outNelems
1756+
inputs[0].scalar_type(), // dType
1757+
std::vector<int64_t>(), // inSplitSizes
1758+
std::vector<int64_t>(), // outSplitSizes
1759+
-1, // globalRankStart
1760+
-1, // globalRankStride
1761+
this->getSize()); // worldSize
1762+
17431763
return collectiveCoalesced(
17441764
inputs,
17451765
outputs,

0 commit comments

Comments
 (0)