已关闭
[Documentation] swin_transformer_ln_qkv: README中beta参数数据类型描述与代码实际支持不符 #2668
daishuangshuang创建于  5月26日关闭于  6月3日
daishuangshuang
5月26日 创建

一、问题描述(必填)

swin_transformer_ln_qkv 算子的 README.md 参数表中,beta 参数的数据类型描述为 "FLOAT16、BFLOAT16、INT8" 三种类型,但实际代码中 beta 仅支持 DT_FLOAT16 一种类型。

二、问题位置

  • 文件:ffn/swin_transformer_ln_qkv/README.md 行 64
  • 关联文件:ffn/swin_transformer_ln_qkv/op_host/swin_transformer_ln_qkv_def.cpp 行 34

三、问题详情

README 描述(行 64):

| beta | 输入 | ... | FLOAT16、BFLOAT16、INT8 | ND |

代码实际定义(def.cpp 行 32-35):

this->Input("beta")
    .ParamType(REQUIRED)
    .DataType({ ge::DT_FLOAT16 })    // 仅支持 FLOAT16
    .Format({ ge::FORMAT_ND })

代码实现中 beta 也只按 FP16 处理,BFLOAT16 和 INT8 均无对应分支。

四、修复建议

将 README 参数表中 beta 的数据类型改为仅 FLOAT16

| beta | 输入 | ... | FLOAT16 | ND |

五、影响范围

文档与代码不一致会误导使用者,可能导致用户传入 BFLOAT16 或 INT8 类型的 beta 后运行失败。

likedislike
huang-chuhong成员
5月26日 评论:

/assign @yolic

likedislike
CANN-robotCANN-robot成员
5月26日 将 yolic 设为负责人
Ddaishuangshuang
5月26日 issue类型由 任务 改变为 缺陷
yuning_chen
yuning_chen成员
5月27日 评论:

您好,感谢您的关注,问题尽快修改合入

likedislike
CANN-robotCANN-robot成员
6月3日 关闭了 issue
CANN-robotCANN-robot成员
6月3日 添加了label:resolved
Ddaishuangshuang
6月8日 issue状态由 待办的 改变为 已完成
CANN-robotCANN-robot成员
6月8日 添加了label:Accepted