Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Chao1Han committed Sep 11, 2024
1 parent df81919 commit b0c0592
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch/csrc/distributed/c10d/ProcessGroupXCCL.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -264,8 +264,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::collective(

work = initWork(device, rank_, opType);

work->outputs_ =
std::make_shared<std::vector<at::Tensor>>(outputs);
work->outputs_ = std::make_shared<std::vector<at::Tensor>>(outputs);
c10::xpu::XPUCachingAllocator::recordStream(
input.storage().data_ptr(), stream);

Expand Down

0 comments on commit b0c0592

Please sign in to comment.