未验证 提交 b8008580 编写于 作者: H houj04 提交者: GitHub

[XPU] update xccl to 1.0.8 and xdnn to 20230215 (#50247)

* [XPU] update xccl to 1.0.8

* update xdnn. add uint8 for concat and split.

* update xdnn to 20230215.
上级 c86a5140
......@@ -7,8 +7,8 @@ set(XPU_PROJECT "extern_xpu")
set(XPU_API_LIB_NAME "libxpuapi.so")
set(XPU_RT_LIB_NAME "libxpurt.so")
set(XPU_BASE_DATE "20230119")
set(XPU_XCCL_BASE_VERSION "1.0.7")
set(XPU_BASE_DATE "20230215")
set(XPU_XCCL_BASE_VERSION "1.0.8")
if(NOT DEFINED XPU_BASE_URL)
set(XPU_BASE_URL_WITHOUT_DATE
......
......@@ -51,10 +51,12 @@ inline BKCLDataType ToBKCLDataType(framework::proto::VarType::Type type) {
return BKCL_FLOAT64;
} else if (type == framework::proto::VarType::FP16) {
return BKCL_FLOAT16;
} else if (type == framework::proto::VarType::UINT8) {
return BKCL_UINT8;
} else {
PADDLE_THROW(platform::errors::Unimplemented(
"BKCL currently only support FP32, INT64, INT32, FP64 and FP16, other "
"data types are not supported."));
"BKCL currently only support FP32, INT64, INT32, FP64, FP16 and UINT8, "
"other data types are not supported."));
}
}
......
......@@ -149,6 +149,9 @@ void ConcatDenseTensorWithType(const phi::XPUContext &dev_ctx,
case phi::DataType::INT64:
ConcatDenseTensor<phi::XPUContext, int64_t>()(dev_ctx, t_list, p_out);
break;
case phi::DataType::UINT8:
ConcatDenseTensor<phi::XPUContext, uint8_t>()(dev_ctx, t_list, p_out);
break;
default:
PADDLE_THROW(platform::errors::Unimplemented(
"Data type (%s) is not supported when it concats tensors.", type));
......@@ -217,6 +220,9 @@ void SplitDenseTensorWithType(const phi::XPUContext &dev_ctx,
case phi::DataType::INT64:
SplitDenseTensor<phi::XPUContext, int64_t>()(dev_ctx, t_in, p_list);
break;
case phi::DataType::UINT8:
SplitDenseTensor<phi::XPUContext, uint8_t>()(dev_ctx, t_in, p_list);
break;
default:
PADDLE_THROW(platform::errors::Unimplemented(
"Data type (%s) is not supported when it splits tensors.", type));
......
......@@ -95,7 +95,7 @@ inline const char* xpuGetErrorString(int stat) {
case XPUERR_INTERRUPTED:
return "Execution interrupted by user";
default:
return "unkonwn error";
return "unknown error";
}
}
......
......@@ -129,6 +129,7 @@ DEFINE_XPU_FUNCTOR(float)
DEFINE_XPU_FUNCTOR(phi::dtype::float16)
DEFINE_XPU_FUNCTOR(int32_t)
DEFINE_XPU_FUNCTOR(int64_t)
DEFINE_XPU_FUNCTOR(uint8_t)
} // namespace funcs
} // namespace phi
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册