issues
search
PaddlePaddle
/
PaddleCustomDevice
PaddlePaddle custom device implementaion. (『飞桨』自定义硬件接入实现)
Apache License 2.0
70
stars
148
forks
source link
[NPU] Fix unittest
#1086
Closed
Tongkaio
closed
7 months ago
Tongkaio
commented
7 months ago
修改:
scale_kernel.cc:ScaleKernel 的参数列表中,bias 的类型已由 float 改为 Scalar,参考:
#62598
;
test_expand_v2_op_npu.py:expand 已支持 uint8 等类型,同步修改单测,参考:
#62849
;
test_transpose_op_npu.py:transpose 已经支持 stride,默认调用 Paddle 主框架中的 transpose,只有 npu 的 transpose 支持非标准格式,故设置 FLAGS_use_stride_kernel 为 False 来调用 npu 上的 transpose。
相关单测已测试通过;
test_elementwise_add_op_npu
test_matmulv2_op_npu
test_npu_identity_op
test_multinomial_op_npu
test_momentum_op_npu
test_rms_norm_npu
test_rope_npu
test_transpose_op_npu
test_unbind_op_npu
test_expand_v2_op_npu
test_flashattention_npu
test_adadelta_op_npu
test_LeNet_MNIST
paddle-bot[bot]
commented
7 months ago
Thanks for your contribution!
修改:
相关单测已测试通过;