From 46c71f2cf6fc03810640202b6ca6956746b640ed Mon Sep 17 00:00:00 2001 From: taixiurong Date: Mon, 29 Nov 2021 19:48:43 +0800 Subject: [PATCH] =?UTF-8?q?DLTP-40731=20[Bug]=20xpu1+x86=E7=8E=AF=E5=A2=83?= =?UTF-8?q?=EF=BC=8Cdevelop=20paddle=E5=8C=85=EF=BC=8Cnlp=20case=20glue=5F?= =?UTF-8?q?xpu1=5Fdy=5Fbert=5Fbs32=20(#37666)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- paddle/fluid/platform/device/xpu/xpu1_op_list.h | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/platform/device/xpu/xpu1_op_list.h b/paddle/fluid/platform/device/xpu/xpu1_op_list.h index 29df763dc0a..1cc7bba132e 100644 --- a/paddle/fluid/platform/device/xpu/xpu1_op_list.h +++ b/paddle/fluid/platform/device/xpu/xpu1_op_list.h @@ -198,6 +198,7 @@ XPUOpMap& get_kl1_ops() { pOpKernelType(vartype::INT64, XPUPlace())})}, {"sgd", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"adam", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, + {"adamw", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"rmsprop", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"lamb", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"pool2d", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, -- GitLab