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

[XPU] fix fallback (#53801)

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