cann-robotfix expint bessel1e fill issue

文件	最后提交记录	最后更新时间
examples	besseli1e算子开发 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !3780 merge besseli1e into master besseli1e算子开发 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2216 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3780	1 个月前
op_graph	besseli1e算子开发 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !3780 merge besseli1e into master besseli1e算子开发 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2216 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3780	1 个月前
op_host	fix expint bessel1e fill issue Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !4439 merge feat/op-check-if-validation into master fix expint bessel1e fill issue Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 描述本 PR 为 `bessel_i1e`、`expint`、`fill_diagonal_v2` 三个算子的 Tiling（host 侧）代码统一补充 `OP_CHECK_IF` 输入合法性校验，修复异常输入无法在 Tiling 阶段提前拦截、报错信息不规范的问题。修改前，上述算子在传入维数超过 8、不支持 dtype、dtype 不一致或输入输出 shape 不匹配时，不会在 Tiling 阶段失败，可能进入非法计算路径导致崩溃或结果错误，且报错信息缺少算子名/输入名/支持类型等可定位字段。 ### 改动内容 #### 1. `conversion/fill_diagonal_v2/op_host/fill_diagonal_v2_tiling.cpp`（+28） - 新增常量 `MAX_DIM_NUM = 8`。 - 在 `Tiling4FillDiagonalV2` 中新增校验： - `x` 维数 `<= 8`，否则 `OP_LOGE_FOR_INVALID_SHAPEDIM_WITH_REASON` 报错并返回 `GRAPH_FAILED`。 - `x` dtype 必须在支持列表 `{DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_UINT8, DT_BOOL, DT_INT8, DT_INT16, DT_INT32, DT_BF16, DT_INT64}` 内，否则 `OP_LOGE_WITH_INVALID_INPUT_DTYPE` 报错。 - `fill_value`（输入 1）dtype 必须与 `x` dtype 一致，否则 `OP_LOGE` 报错。 #### 2. `math/bessel_i1e/op_host/arch35/bessel_i1e_tiling_arch35.cpp`（+12/-1） - 新增常量 `MAX_DIM_NUM = 8`。 - 在 `GetShapeAttrsInfo` 中新增 `x` 维数 `<= 8` 校验。 - 将原有 `OP_LOGE(context, "invalid dtype")` 替换为 `OP_LOGE_WITH_INVALID_INPUT_DTYPE`，报错包含算子名、输入名、实际 dtype 及支持列表 `DT_FLOAT, DT_FLOAT16, DT_BFLOAT16`。 #### 3. `math/expint/op_host/arch35/expint_tiling_arch35.cpp`（+19/-1） - 新增常量 `MAX_DIM_NUM = 8`。 - 在 `GetShapeAttrsInfo` 中新增校验： - `x` 维数 `<= 8`。 - 输入 `x` 与输出 `y` 的 shape size 一致，否则 `OP_LOGE` 报错。 - 将原有 `OP_LOGE(context, "Expint: unsupported dtype=%d", ...)` 替换为 `OP_LOGE_WITH_INVALID_INPUT_DTYPE`，报错包含算子名、输入名、实际 dtype 及支持列表 `DT_FLOAT16, DT_FLOAT, DT_BFLOAT16`。 #### 4. `conversion/fill_diagonal_v2/tests/ut/op_host/test_fill_diagonal_v2_tiling.cpp`（+34/-31） - 调整 UT 代码格式以通过 pre-commit 检查（`test_fill_diagonal_v2_float` 中第二个输入由 `DT_FLOAT16` 改为 `DT_FLOAT`，与新增加的 dtype 一致性校验保持一致；统一代码风格）。 ### 改动方法说明统一采用 `OP_CHECK_IF(condition, error_log, return_statement)` 模式进行校验，配合规范化报错宏： - `OP_LOGE_FOR_INVALID_SHAPEDIM_WITH_REASON`：用于 shape 维数类错误，输出算子名、输入名、实际维数及原因。 - `OP_LOGE_WITH_INVALID_INPUT_DTYPE`：用于 dtype 类错误，输出算子名、输入名、实际 dtype 及支持列表。 - `OP_LOGE`：用于其他自定义校验（如 shape size 一致性、dtype 一致性）。 ## 关联的 Issue - 见 `pr4439_issue.md`：bessel_i1e / expint / fill_diagonal_v2 算子 Tiling 缺少输入校验。 ## 测试 - `fill_diagonal_v2` UT：更新 `test_fill_diagonal_v2_tiling.cpp`，3 个用例（float / float_cube / float16）执行通过，校验新增的 dtype 一致性约束不破坏既有用例。 - CI：`ci-pipeline-passed`、`api-check-passed` 标签已通过。 - `bessel_i1e`、`expint`：本仓库无对应 UT 目录，通过二级冒烟/算子泛化验证。 ## 文档更新无。本次仅修改 Tiling host 代码与 UT，不涉及对外接口或 README 变更。 ## 类型标签 - [x] Bug 修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## 提交记录 1. `fix issue` — 初始修复 2. `revert: 只保留OP_CHECK_IF相关修改，恢复ASCENDC_TPL_SEL_PARAM等非OP_CHECK_IF改动` — 回退非 OP_CHECK_IF 改动，聚焦校验逻辑 3. `feat: add OP_CHECK_IF validation for bessel_i1e/expint/fill_diagonal_v2 tiling` — 主功能提交 4. `fix: 修复 fill_diagonal_v2 tiling ut 代码格式以通过 pre-commit 检查` — 格式修复 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2532 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4439	1 个月前
op_kernel	bessel_i1e fix bug Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !4131 merge besseli1e into master bessel_i1e fix bug Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2271 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4131	1 个月前
tests	besseli1e算子开发 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !3780 merge besseli1e into master besseli1e算子开发 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2216 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3780	1 个月前
CMakeLists.txt	besseli1e算子开发 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !3780 merge besseli1e into master besseli1e算子开发 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2216 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3780	1 个月前
README.md	besseli1e算子开发 Co-authored-by: kangjiaming<1159380836@qq.com> # message auto-generated for no-merge-commit merge: !3780 merge besseli1e into master besseli1e算子开发 Created-by: kangjiaming Commit-by: kangjiaming Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> https://gitcode.com/cann/ops-math/issues/2216 <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3780	1 个月前

BesselI1e

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：计算指数缩放的第一阶修正贝塞尔函数。
计算公式：

outi=e−∣inputi∣⋅I1(inputi)out_i = e^{-|input_i|} \cdot I_1(input_i)

其中 $I_1(x)$ 是第一阶修正贝塞尔函数。指数缩放因子 $e^{-|x|}$ 使得函数在大参数值时数值稳定。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行BesselI1e计算的入参。	FLOAT、FLOAT16、BF16	ND
y	输出	BesselI1e计算的出参。	FLOAT、FLOAT16、BF16	ND

约束说明

无

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_bessel_i1e	通过算子IR构图方式调用BesselI1e算子。