Skip to content

Commit e6fdcd9

Browse files
authored
xpu support inplace flatten (#48909)
This is a PR to catch up with latest xpu white list strategy (#48606) , since original list only include 'fluid' fashion names, but new list must include 'phi' fashion as well. Refer to paddle/phi/core/kernel_factory.cc for more details.
1 parent 905be66 commit e6fdcd9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

paddle/phi/backends/xpu/xpu2_op_list.cc

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,12 @@ XPUOpMap& get_kl2_ops() {
227227
phi::DataType::INT8,
228228
phi::DataType::FLOAT16,
229229
phi::DataType::FLOAT32})},
230+
{"flatten_with_xshape",
231+
XPUKernelSet({phi::DataType::INT64,
232+
phi::DataType::INT32,
233+
phi::DataType::INT8,
234+
phi::DataType::FLOAT16,
235+
phi::DataType::FLOAT32})},
230236
{"flatten_grad",
231237
XPUKernelSet({phi::DataType::INT64,
232238
phi::DataType::INT32,

0 commit comments

Comments
 (0)