已关闭
[Bug-Report|缺陷反馈]: 与canndev对齐fused_mul_apply_momentum_extern参数类型,并取消BF16支持 #5132
牛康创建于  8月25日关闭于  8月26日
牛康
牛康
8月25日 创建

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 / 必填)

测试通过

不涉及

Special notes for this issue/备注 (Optional / 选填)

likedislike
牛康牛康
8月25日 关联了pull request:fix: correct parameter type in proto for fused_mul_apply_momentum_extern
牛康牛康
8月25日 修改了issue 的描述
牛康牛康
8月25日 修改了issue 的描述
yuning_chenyuning_chen成员
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日 修改了issue 的描述
牛康牛康
8月26日 修改了issue 的描述
牛康牛康
8月26日 修改了issue 的描述
牛康牛康
8月26日 修改了issue 的描述
CANN-robotCANN-robot成员
8月26日 关闭了 issue
CANN-robotCANN-robot成员
8月26日 添加了label:resolved