From b7bafee81026be15dc996a9640f222844c63a8b1 Mon Sep 17 00:00:00 2001
From: houj04 <35131887+houj04@users.noreply.github.com>
Date: Thu, 23 Dec 2021 15:12:23 +0800
Subject: [PATCH] add-leaky-relu-to-xpu2-op-list (#38366)

---
 paddle/fluid/platform/device/xpu/xpu2_op_list.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/platform/device/xpu/xpu2_op_list.h b/paddle/fluid/platform/device/xpu/xpu2_op_list.h
index 74f519c7a86..255c818b98c 100644
--- 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())})},
-- 
GitLab