add fp16 for reshape op on kunlun2, *test=kunlun (#42605)

754edf6e · TTerror · GitHub · 668a0a41 · 754edf6e
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/platform/device/xpu/xpu2_op_list.h paddle/fluid/platform/device/xpu/xpu2_op_list.h +2 -0

未找到文件。
--- a/paddle/fluid/platform/device/xpu/xpu2_op_list.h
+++ b/paddle/fluid/platform/device/xpu/xpu2_op_list.h
@@ -307,11 +307,13 @@ XPUOpMap& get_kl2_ops() {
                             pOpKernelType(vartype::FP16, XPUPlace())})},
      {"reshape2_grad",
       XPUKernelSet({pOpKernelType(vartype::FP64, XPUPlace()),
+                     pOpKernelType(vartype::FP16, XPUPlace()),
                     pOpKernelType(vartype::INT64, XPUPlace()),
                     pOpKernelType(vartype::INT32, XPUPlace()),
                     pOpKernelType(vartype::BOOL, XPUPlace()),
                     pOpKernelType(vartype::FP32, XPUPlace())})},
      {"reshape2", XPUKernelSet({pOpKernelType(vartype::FP64, XPUPlace()),
+                                 pOpKernelType(vartype::FP16, XPUPlace()),
                                 pOpKernelType(vartype::INT64, XPUPlace()),
                                 pOpKernelType(vartype::INT32, XPUPlace()),
                                 pOpKernelType(vartype::BOOL, XPUPlace()),