[XPU] add load op into oplist. (#48860)

* [XPU] add load op into oplist. * remove test_sampling_id_op_xpu.py

[XPU] add load op into oplist. (#48860)
* [XPU] add load op into oplist. * remove test_sampling_id_op_xpu.py
2bba3e18 · houj04 · GitHub · 782925d4 · 2bba3e18
显示空白变更内容
内联并排

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
@@ -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,