| Files | Last commit | Last update |
|---|---|---|
fix(TopKFusionPass): data_type DT_INT64 of input[1, k] is unsupported Co-authored-by: jerry_gd<jerry.luo@huawei.com> # message auto-generated for no-merge-commit merge: !7006 merge TopKFusionPass into master fix(TopKFusionPass): data_type DT_INT64 of input[1, k] is unsupported Created-by: jerry_gd Commit-by: jerry_gd Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 本 PR 修复TopKFusionPass 在两处场景下的 dtype 处理问题。其一，被各道门禁拒绝（x dtype 不合规、sorted=false、TopKD 后端拒绝）的 TopK 源节点如果 k 输入 desc 为 int64，原实现直接跳过，导致后续 AICPU 选型报 DT_INT64 ... unsupported；现以 int32 的 k desc 重建该节点。其二，AICPU 降级替换时 indices 输出 desc 改为无条件声明为 int32，避免从 values desc 继承 double 等不可选型 dtype。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/4413 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> - 二级冒烟通过。 - ascend310B芯片下，跑相关k为int64的用例通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及。 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!7006 | 12 days ago |