search

PaddlePaddle / PaddleCustomDevice

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

[NPU] fix static kernel. #1066

Closed xuanyuanminzheng closed 7 months ago

xuanyuanminzheng commented 7 months ago

fix static kernel: (1) slice_kernel.cc -> SliceGradRawKernel -> padD, unit test: test_slice_op_npu.py (2) index_sample_kernel.cc -> IndexSampleGather-> concat(D), unit test: test_concat_op_npu.py

add bf16 support and unit test: pad_kernel.cc -> test_pad_op_npu_eager.py

paddle-bot[bot] commented 7 months ago

Thanks for your contribution!