DLTP-40731 [Bug] xpu1+x86环境，develop paddle包，nlp case glue_xpu1_dy_bert_bs32 (#37666)

46c71f2c · taixiurong · GitHub · f9e9fd19 · 46c71f2c
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/platform/device/xpu/xpu1_op_list.h paddle/fluid/platform/device/xpu/xpu1_op_list.h +1 -0

未找到文件。
--- 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())})},