已开启
[Documentation|文档反馈]: bessel_i0e README 漏列 DOUBLE,与 def/kernel dtype 声明不一致 #4
xujiachen8创建于  9月14日
xujiachen8成员
9月14日 创建

本 Issue 仅用于 gitcode-issue-handler 调试。原始来源:https://gitcode.com/cann/ops-math/issues/2926 。以下保留原始描述。

在您提交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。

likedislike
xujiachen8成员
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 后继续产生注册声明与运行时行为不一致。

likedislike