Skip to content

Commit 664b9d8

Browse files
committed
update
1 parent 336a74b commit 664b9d8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/xccl/ProcessGroupXCCL.hpp

+3-2
Original file line numberDiff line numberDiff line change
@@ -219,8 +219,9 @@ class TORCH_API ProcessGroupXCCL : public Backend {
219219
const char* profilingTitle = nullptr);
220220

221221
c10::intrusive_ptr<Work> allreduce_impl(
222-
at::Tensor& tensor,
223-
const AllreduceOptions& opts = AllreduceOptions());
222+
at::Tensor& tensor,
223+
const char* profilingTitle = "xccl:all_reduce",
224+
const AllreduceOptions& opts = AllreduceOptions());
224225

225226
c10::intrusive_ptr<Work> allreduce(
226227
std::vector<at::Tensor>& tensors,

0 commit comments

Comments
 (0)