Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 89f903da 编写于 作者: R ronnywang 提交者： GitHub
浏览文件
操作

fix npu c_allgather int64 (#39099)

上级 4adeff06
隐藏空白更改
内联 并排
Showing with 3 addition and 0 deletion
paddle/fluid/operators/collective/c_allgather_op_npu.cc
浏览文件 @ 89f903da
...@@ -79,5 +79,6 @@ namespace plat = paddle::platform; ...@@ -79,5 +79,6 @@ namespace plat = paddle::platform;
REGISTER_OP_NPU_KERNEL(c_allgather, ops::CAllGatherOpASCENDKernel<int8_t>, REGISTER_OP_NPU_KERNEL(c_allgather, ops::CAllGatherOpASCENDKernel<int8_t>,
ops::CAllGatherOpASCENDKernel<int>, ops::CAllGatherOpASCENDKernel<int>,
ops::CAllGatherOpASCENDKernel<int64_t>,
ops::CAllGatherOpASCENDKernel<float>, ops::CAllGatherOpASCENDKernel<float>,
ops::CAllGatherOpASCENDKernel<plat::float16>); ops::CAllGatherOpASCENDKernel<plat::float16>);
paddle/fluid/platform/device/npu/hccl_helper.h
浏览文件 @ 89f903da
...@@ -41,6 +41,8 @@ inline HcclDataType ToHCCLDataType(framework::proto::VarType::Type type) { ...@@ -41,6 +41,8 @@ inline HcclDataType ToHCCLDataType(framework::proto::VarType::Type type) {
return HCCL_DATA_TYPE_FP32; return HCCL_DATA_TYPE_FP32;
} else if (type == framework::proto::VarType::FP16) { } else if (type == framework::proto::VarType::FP16) {
return HCCL_DATA_TYPE_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) { } else if (type == framework::proto::VarType::INT32) {
return HCCL_DATA_TYPE_INT32; return HCCL_DATA_TYPE_INT32;
} else if (type == framework::proto::VarType::INT8) { } else if (type == framework::proto::VarType::INT8) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或