未验证 提交 e6fdcd90 编写于 作者: J james 提交者: GitHub

xpu support inplace flatten (#48909)

This is a PR to catch up with latest xpu white list strategy
(https://github.com/PaddlePaddle/Paddle/pull/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.
上级 905be668
...@@ -227,6 +227,12 @@ XPUOpMap& get_kl2_ops() { ...@@ -227,6 +227,12 @@ XPUOpMap& get_kl2_ops() {
phi::DataType::INT8, phi::DataType::INT8,
phi::DataType::FLOAT16, phi::DataType::FLOAT16,
phi::DataType::FLOAT32})}, phi::DataType::FLOAT32})},
{"flatten_with_xshape",
XPUKernelSet({phi::DataType::INT64,
phi::DataType::INT32,
phi::DataType::INT8,
phi::DataType::FLOAT16,
phi::DataType::FLOAT32})},
{"flatten_grad", {"flatten_grad",
XPUKernelSet({phi::DataType::INT64, XPUKernelSet({phi::DataType::INT64,
phi::DataType::INT32, phi::DataType::INT32,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册