diff --git a/paddle/phi/backends/xpu/xpu2_op_list.cc b/paddle/phi/backends/xpu/xpu2_op_list.cc
index 8b19d3a5c4efaa21d52ea590174ab529f1f643c6..cd68540017001cc0575ad5a86c2e1a4f0cbab3da 100644
--- a/paddle/phi/backends/xpu/xpu2_op_list.cc
+++ b/paddle/phi/backends/xpu/xpu2_op_list.cc
@@ -300,6 +300,7 @@ XPUOpMap& get_kl2_ops() {
        XPUKernelSet({phi::DataType::INT64,
                      phi::DataType::INT32,
                      phi::DataType::FLOAT32})},
+      {"load", XPUKernelSet({phi::DataType::FLOAT32})},
       {"load_combine",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT64,