Skip to content

Commit 7eb17ff

Browse files
Chao1HanCopilot
andauthored
Fix dist.gather with noncontiguous tensor (#1903)
Refer to pytorch/pytorch#159548, XCCL may have same issue. --------- Co-authored-by: Copilot <[email protected]>
1 parent a405a3e commit 7eb17ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/xccl/ProcessGroupXCCL.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -865,7 +865,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::gather(
865865
TORCH_CHECK(inputTensors.size() == 1, MULTI_DEVICE_ERROR_MSG);
866866
// @lint-ignore CLANGTIDY
867867
auto inputTensor = inputTensors.back();
868-
868+
checkSingleTensor(inputTensor);
869869
std::vector<at::Tensor> outputs;
870870

871871
if (getRank() == opts.rootRank) {

0 commit comments

Comments
 (0)