search

PaddlePaddle / PaddleCustomDevice

PaddlePaddle custom device implementaion. (『飞桨』自定义硬件接入实现)
Apache License 2.0
70 stars 148 forks source link

[NPU] add aclnn of AllRawKernel #1060

Closed caijuanjuan closed 7 months ago

caijuanjuan commented 7 months ago

本次迁移kernel：AllRawKernel、AllKernel，对应的测试用例：test_reduce_all_op_npu expand_as中调用到了AllKernel，调Aclop时不出错，但调aclnn时报错，主要是因为，aclnn的out输出，需要size为[[]]的tensor，但expand_as中构建了一个size为[[1]]的tensor，因此修改expand_as中对out进行resize的代码

paddle-bot[bot] commented 7 months ago

Thanks for your contribution!