未验证 提交 4b85e5db 编写于 作者: W wz1qqx 提交者: GitHub

[XPU] fix fallback (#53801)

上级 97690816
...@@ -187,7 +187,8 @@ XPUOpMap& get_kl2_ops() { ...@@ -187,7 +187,8 @@ XPUOpMap& get_kl2_ops() {
{"depthwise_conv2d", XPUKernelSet({phi::DataType::FLOAT32})}, {"depthwise_conv2d", XPUKernelSet({phi::DataType::FLOAT32})},
{"depthwise_conv2d_transpose_grad", {"depthwise_conv2d_transpose_grad",
XPUKernelSet({phi::DataType::FLOAT32})}, XPUKernelSet({phi::DataType::FLOAT32})},
{"depthwise_conv2d_transpose", XPUKernelSet({phi::DataType::FLOAT32})}, {"depthwise_conv2d_transpose",
XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
{"diag_v2", {"diag_v2",
XPUKernelSet({phi::DataType::FLOAT32, XPUKernelSet({phi::DataType::FLOAT32,
phi::DataType::FLOAT16, phi::DataType::FLOAT16,
...@@ -709,7 +710,8 @@ XPUOpMap& get_kl2_ops() { ...@@ -709,7 +710,8 @@ XPUOpMap& get_kl2_ops() {
{"split", {"split",
XPUKernelSet({phi::DataType::FLOAT32, XPUKernelSet({phi::DataType::FLOAT32,
phi::DataType::FLOAT16, phi::DataType::FLOAT16,
phi::DataType::INT32})}, phi::DataType::INT32,
phi::DataType::INT64})},
{"split_with_num", {"split_with_num",
XPUKernelSet({phi::DataType::FLOAT32, XPUKernelSet({phi::DataType::FLOAT32,
phi::DataType::FLOAT16, phi::DataType::FLOAT16,
......
...@@ -177,7 +177,8 @@ PD_REGISTER_KERNEL(depthwise_conv2d_transpose, ...@@ -177,7 +177,8 @@ PD_REGISTER_KERNEL(depthwise_conv2d_transpose,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DepthwiseConv2dTransposeKernel, phi::DepthwiseConv2dTransposeKernel,
float) {} float,
phi::dtype::float16) {}
PD_REGISTER_KERNEL(conv2d_transpose, PD_REGISTER_KERNEL(conv2d_transpose,
XPU, XPU,
......
...@@ -66,9 +66,14 @@ void SplitWithNumKernel(const Context& dev_ctx, ...@@ -66,9 +66,14 @@ void SplitWithNumKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PD_REGISTER_KERNEL( PD_REGISTER_KERNEL(split,
split, XPU, ALL_LAYOUT, phi::SplitKernel, float, int, phi::dtype::float16) { XPU,
} ALL_LAYOUT,
phi::SplitKernel,
float,
int64_t,
int,
phi::dtype::float16) {}
PD_REGISTER_KERNEL(split_with_num, PD_REGISTER_KERNEL(split_with_num,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册