add-leaky-relu-to-xpu2-op-list (#38366)

b7bafee8 · houj04 · GitHub · f88065d3 · b7bafee8
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

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

未找到文件。
--- a/paddle/fluid/platform/device/xpu/xpu2_op_list.h
+++ b/paddle/fluid/platform/device/xpu/xpu2_op_list.h
@@ -194,6 +194,9 @@ XPUOpMap& get_kl2_ops() {
      {"layer_norm", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
      {"layer_norm", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace()),
                                   pOpKernelType(vartype::FP16, XPUPlace())})},
+      {"leaky_relu_grad",
+       XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
+      {"leaky_relu", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
      {"less_equal", XPUKernelSet({pOpKernelType(vartype::INT64, XPUPlace()),
                                   pOpKernelType(vartype::INT32, XPUPlace()),
                                   pOpKernelType(vartype::FP32, XPUPlace())})},