fix xpu fp16 lod_reset (#52346)

b4137338 · zhupengyang · GitHub · 50c949f0 · b4137338
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/phi/backends/xpu/xpu2_op_list.cc paddle/phi/backends/xpu/xpu2_op_list.cc +1 -0

未找到文件。
--- a/paddle/phi/backends/xpu/xpu2_op_list.cc
+++ b/paddle/phi/backends/xpu/xpu2_op_list.cc
@@ -888,6 +888,7 @@ XPUOpMap& get_kl2_ops() {
       XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
      {"lod_reset",
       XPUKernelSet({phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT16,
                     phi::DataType::FLOAT64,
                     phi::DataType::INT32,
                     phi::DataType::INT64})},