已开启
[Documentation|文档反馈]: bessel_i0e README 漏列 DOUBLE,与 def/kernel dtype 声明不一致 #4
xujiachen8创建于 9月14日
xujiachen8
9月16日 评论:
9月16日 评论:
您好,感谢反馈!
问题分析
math/bessel_i0e/README.md 的 x、y 数据类型确实漏列了 DOUBLE:算子定义 bessel_i0e_def.cpp 的输入和输出声明包含 DT_DOUBLE,arch35 bessel_i0e_tiling_key.h 也包含 C_DT_DOUBLE。
同时我们核对到实现层还有一处需要一起统一的范围差异:arch35 tiling 的 supportedDtypes 和错误提示目前只接受 DT_FLOAT16、DT_FLOAT、DT_BF16,DT_DOUBLE 会在 tiling 阶段被拒绝;kernel 的说明和转换分支也按 FP32/FP16/BF16 编写。当前没有找到明确关联来源 Issue #2926、且覆盖这组 dtype 声明的修复 PR。
处理方向
README 的类型表需要与实际可用范围保持一致。维护侧会沿 OpDef、arch35 tiling dtype 校验、kernel/图模式类型约束和验证用例统一确认 DOUBLE 的最终支持范围,再同步文档,避免只修改 README 后继续产生注册声明与运行时行为不一致。


在您提交issue前,请确认以下信息:
问题描述
math/bessel_i0e/README.md的 x/y 数据类型只列了FLOAT16、FLOAT、BF16(:46、:53),但算子定义与 kernel 模板实际支持 DOUBLE:op_host/bessel_i0e_def.cpp:28/33:DataType 为DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_BF16。op_kernel/arch35/bessel_i0e_tiling_key.h:18-24:kernel 模板 dtype 选择包含C_DT_DOUBLE。建议修复
README 两处改为
FLOAT16、FLOAT、DOUBLE、BF16。