Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
shape:"((14, 16, 10, 12, 16), (11,), (11, 11, 10, 12, 16), (16,), (16,))" dtype:"('float8_e5m2', 'int32', 'bf16', 'bf16', 'bf16')"
Ascend950
QuantUpdateScatter 执行用例比对失败
精度ok
无
/assign
indices是1维时,模板index/quant_update_scatter/op_kernel/arch35/quant_update_scatter_large_batch_little_quant_regbase.h的output的offset计算有误,需要修改
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
shape:"((14, 16, 10, 12, 16), (11,), (11, 11, 10, 12, 16), (16,), (16,))"
dtype:"('float8_e5m2', 'int32', 'bf16', 'bf16', 'bf16')"
Environment / 环境信息 (Mandatory / 必填)
Ascend950
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
QuantUpdateScatter 执行用例比对失败
Describe the expected behavior / 预期结果 (Mandatory / 必填)
精度ok
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
无
Special notes for this issue/备注 (Optional / 选填)