PaddlePaddle / PaddleCustomDevice

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

support argmax aclnn kernel #1077

Closed warrentdrew closed 7 months ago

warrentdrew commented 7 months ago

support aclnn for argmax kernel unittest: test_arg_max_op_npu.py

paddle-bot[bot] commented 7 months ago

Thanks for your contribution!