Closed qili93 closed 5 months ago
CANN 8.0.RC1 存在BUG,根据华为昇腾同学建议,商发包修改下这段代码
文件: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)
Thanks for your contribution!
CANN 8.0.RC1 存在BUG,根据华为昇腾同学建议,商发包修改下这段代码
文件: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)