Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Chao1Han committed Oct 21, 2024
1 parent ea4fe8c commit e36a99c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch/csrc/distributed/c10d/ProcessGroupXCCL.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1060,7 +1060,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::_allgather_base(
TORCH_CHECK_WITH(
TypeError,
input_tensor.dtype() == output_tensor.dtype(),
"output tensor must have the same type as input tensor");
"input tensor must be the same type as the output tensor.");
TORCH_CHECK_WITH(
ValueError,
input_tensor.numel() * size_ == output_tensor.numel(),
Expand Down Expand Up @@ -1189,7 +1189,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::_reduce_scatter_base(
TORCH_CHECK_WITH(
TypeError,
inputTensor.dtype() == outputTensor.dtype(),
"output tensor must have the same type as input tensor");
"input tensor must be the same type as the output tensor.");
TORCH_CHECK_WITH(
ValueError,
inputTensor.numel() == outputTensor.numel() * size_,
Expand Down

0 comments on commit e36a99c

Please sign in to comment.