From 2bba3e187663e48af2bcf94ddfd34b324ea49c9a Mon Sep 17 00:00:00 2001
From: houj04 <35131887+houj04@users.noreply.github.com>
Date: Thu, 8 Dec 2022 15:07:49 +0800
Subject: [PATCH] [XPU] add load op into oplist. (#48860)

* [XPU] add load op into oplist.

* remove test_sampling_id_op_xpu.py
---
 paddle/phi/backends/xpu/xpu2_op_list.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/phi/backends/xpu/xpu2_op_list.cc b/paddle/phi/backends/xpu/xpu2_op_list.cc
index 8b19d3a5c4e..cd685400170 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,
-- 
GitLab