We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 336a74b commit 664b9d8Copy full SHA for 664b9d8
src/xccl/ProcessGroupXCCL.hpp
@@ -219,8 +219,9 @@ class TORCH_API ProcessGroupXCCL : public Backend {
219
const char* profilingTitle = nullptr);
220
221
c10::intrusive_ptr<Work> allreduce_impl(
222
- at::Tensor& tensor,
223
- const AllreduceOptions& opts = AllreduceOptions());
+ at::Tensor& tensor,
+ const char* profilingTitle = "xccl:all_reduce",
224
+ const AllreduceOptions& opts = AllreduceOptions());
225
226
c10::intrusive_ptr<Work> allreduce(
227
std::vector<at::Tensor>& tensors,
0 commit comments