Skip to content

Commit 2bba3e1

Browse files
authored
[XPU] add load op into oplist. (#48860)
* [XPU] add load op into oplist. * remove test_sampling_id_op_xpu.py
1 parent 782925d commit 2bba3e1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddle/phi/backends/xpu/xpu2_op_list.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -300,6 +300,7 @@ XPUOpMap& get_kl2_ops() {
300300
XPUKernelSet({phi::DataType::INT64,
301301
phi::DataType::INT32,
302302
phi::DataType::FLOAT32})},
303+
{"load", XPUKernelSet({phi::DataType::FLOAT32})},
303304
{"load_combine",
304305
XPUKernelSet({phi::DataType::FLOAT32,
305306
phi::DataType::FLOAT64,

0 commit comments

Comments
 (0)