PaddlePaddle / PaddleCustomDevice

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

[NPU] fix CANN 8.0.RC1 bug #1261

Closed qili93 closed 5 months ago

qili93 commented 5 months ago

CANN 8.0.RC1 存在BUG,根据华为昇腾同学建议,商发包修改下这段代码

image

文件:ops/built-in/tbe/impl/dynamic/rotary_mul_grad.py

行数:205行

self.burst.set_as((self.dim_3 + self.x_data_each_block - 1) // self.x_data_each_block)

-->

self.burst.set_as(

(self.ub_per_reser_number * self.dim_3 + self.x_data_each_block - 1) // self.x_data_each_block)

paddle-bot[bot] commented 5 months ago

Thanks for your contribution!