diff --git a/paddle/fluid/operators/collective/c_allgather_op_npu.cc b/paddle/fluid/operators/collective/c_allgather_op_npu.cc index 4fa27f5eb9bee61f3ab7684888c49c06e550855d..5ebcc9064f79062c5e07bddfd9c375de025b70ee 100644 --- a/paddle/fluid/operators/collective/c_allgather_op_npu.cc +++ b/paddle/fluid/operators/collective/c_allgather_op_npu.cc @@ -79,5 +79,6 @@ namespace plat = paddle::platform; REGISTER_OP_NPU_KERNEL(c_allgather, ops::CAllGatherOpASCENDKernel, ops::CAllGatherOpASCENDKernel, + ops::CAllGatherOpASCENDKernel, ops::CAllGatherOpASCENDKernel, ops::CAllGatherOpASCENDKernel); diff --git a/paddle/fluid/platform/device/npu/hccl_helper.h b/paddle/fluid/platform/device/npu/hccl_helper.h index c2338fff02926cb544f99ad81285e1cb95938612..efbc56bee720b0b5ed4e6aa91d6a7ad0ad14b197 100644 --- a/paddle/fluid/platform/device/npu/hccl_helper.h +++ b/paddle/fluid/platform/device/npu/hccl_helper.h @@ -41,6 +41,8 @@ inline HcclDataType ToHCCLDataType(framework::proto::VarType::Type type) { return HCCL_DATA_TYPE_FP32; } else if (type == framework::proto::VarType::FP16) { return HCCL_DATA_TYPE_FP16; + } else if (type == framework::proto::VarType::INT64) { + return HCCL_DATA_TYPE_INT64; } else if (type == framework::proto::VarType::INT32) { return HCCL_DATA_TYPE_INT32; } else if (type == framework::proto::VarType::INT8) {