未验证 提交 1c1b487c 编写于 作者: L lj970926 提交者: GitHub

[XPU][BUG] Add cumsum grad kernel to xpu2 op list (#53386)

* clang format

* add cumsum_grad op to xpu2_op_list
上级 98fc4277
...@@ -169,6 +169,10 @@ XPUOpMap& get_kl2_ops() { ...@@ -169,6 +169,10 @@ XPUOpMap& get_kl2_ops() {
phi::DataType::FLOAT16, phi::DataType::FLOAT16,
phi::DataType::INT32, phi::DataType::INT32,
phi::DataType::INT64})}, phi::DataType::INT64})},
{"cumsum_grad",
XPUKernelSet({phi::DataType::FLOAT32,
phi::DataType::INT32,
phi::DataType::INT64})},
{"cumprod", {"cumprod",
XPUKernelSet({phi::DataType::FLOAT32, XPUKernelSet({phi::DataType::FLOAT32,
phi::DataType::INT32, phi::DataType::INT32,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册