已关闭
[Bug-Report|缺陷反馈]: 与canndev对齐fused_mul_apply_momentum_extern参数类型,并取消BF16支持 #5132
牛康创建于 8月25日关闭于 8月26日
8月25日 关联了pull request:fix: correct parameter type in proto for fused_mul_apply_momentum_extern
8月25日 修改了issue 的描述
8月25日 修改了issue 的描述
8月25日 将 niukang_hw 设为负责人
8月26日 修改标题为 “[Bug-Report|缺陷反馈]: 与canndev对齐fused_mul_apply_momentum_extern参数类型,并取消BF16支持”,原标题为“[Bug-Report|缺陷反馈]: fused_mul_apply_momentum_extern算子proto中参数类型不为NumberType”
8月26日 修改标题为 “[Bug-Report|缺陷反馈]: 与canndev对齐fused_mul_apply_momentum_extern参数类型,并取消BF16支持”,原标题为“[Bug-Report|缺陷反馈]: fused_mul_apply_momentum_extern算子proto中参数类型不为NumberType”
8月26日 修改了issue 的描述
8月26日 修改了issue 的描述
8月26日 修改了issue 的描述
8月26日 修改了issue 的描述
8月26日 关闭了 issue
8月26日 添加了label:resolved
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
问题现象
测试报错:The op do not support thr dtype: DT_BFLOAT16
表明该算子当前无法处理 BF16(bfloat16)数据类型输入。
根因定位
Kernel 层面:该算子的 kernel 实现中已包含对 BF16 类型的支持。
Proto/注册层面:算子的 Proto 定义(或 REG_OP 参数类型声明)中并未包含 BF16 类型,导致类型检查失败。
对齐要求:该 Proto 定义必须与上游 canndev 仓库保持一致。而 canndev 中该算子不支持 BF16,因此不能独立添加 BF16 支持。
修改方案
fused_mul_apply_momentum_extern 算子删除所有 BF16 相关代码
Environment / 环境信息 (Mandatory / 必填)
不涉及
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
python3 -m ttk geir -i /home/w00509544/test-bot-dev_0807/case/GEIR/FusedMulApplyMomentumExtern/FusedMulApplyMomentumExtern_l0_functional.csv -d -b -c=false --golden-mode Disable -t FusedMulApplyMomentumExtern_L0_009
执行报错:
The op do not support thr dtype: DT_BFLOAT16
Describe the expected behavior / 预期结果 (Mandatory / 必填)
测试通过
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
不涉及
Special notes for this issue/备注 (Optional / 选填)