已关闭
[Bug-Report|缺陷反馈]: TransformBiasRescaleQkv 需要增加num_heads!=0的拦截校验 #2835
guijianwei创建于 23 天前关闭于 22 天前
guijianwei
23 天前 评论：
23 天前 评论：
/assign
23 天前 将 guijianwei 设为负责人
23 天前 关联了pull request：add divide zero check
22 天前 关联了pull request：fix divide zero bug
22 天前 issue状态由 进行中 改变为 已完成
22 天前 关闭了 issue
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
一、问题描述 （必填）
文件:
ops-math/math/transform_bias_rescale_qkv/op_host/transform_bias_rescale_qkv_infershape.cpp漏洞代码 (行75):
q_shape->SetDim(OUTPUT_QKV_DIM3, qkv_shape->GetDim(INPUT_QKV_DIM2) / QKV_NUM / num_heads); // ❌ num_heads=0时除零对，加个除零校验
二、环境信息 （可选）
三、重现步骤 （可选）
四、预期结果 （可选）
💡 备注（选填）