Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Chao1Han committed Dec 30, 2024
1 parent 3d94dd5 commit 0002a7b
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/xccl/ProcessGroupXCCL.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1685,8 +1685,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::alltoall_base(
auto xcclDataType = getXcclDataType(output.scalar_type());
size_t size = input.element_size();

ccl:
group_start();
ccl::group_start();
for (const auto r : c10::irange(size_)) {
if (send_lengths[r] != 0) {
ccl::send(
Expand Down Expand Up @@ -1756,7 +1755,7 @@ c10::intrusive_ptr<Work> ProcessGroupXCCL::alltoall(
at::Tensor& /* unused */,
xcclComm_t& comm,
at::xpu::XPUStream& stream) {
auto xcclDataType = getXcclDataType(input.scalar_type());
auto xcclDataType = getXcclDataType(inputTensors[0].scalar_type());

ccl::group_start();
for (const int r :
Expand Down

0 comments on commit 0002a7b

Please sign in to comment.