Closed
[Bug-Report|缺陷反馈]:ascend310B的TopKFusionPass融合规则报错 #4413
jerry_gdAdd on  7 days agoClosed at  7 days ago
jerry_gdCollaborators
7 days ago Create

一、问题描述 (必填)

ascend310B TopKFusionPass融合规则执行后返回 data_type DT_INT64 of input[1, k] is unsupported

[PID: 876178] 2026-11-02-14:48:37.310.369 Not_Supported_Operator(EZ3004): The operator information library aicpu_ascend_kernel does not support the operator TopK_1265 (operator type: TopK). Cause: data_type DT_INT64 of input[1, k] is unsupported
        Solution: 1. If a built-in operator is used, upgrade to the CANN operator package that supports this specification. 2. If a custom operator is used, register and use the custom operator with the required specification.

二、环境信息 (可选)

ascend310B

三、重现步骤 (可选)

四、预期结果 (可选)

ONNX 规范中 TopK 的 k 是 int64,TopKFusionPass处理ascend310B的TopK节点时,aicpu TopK不支持int64,需要转化。

💡 备注(选填)

likedislike
Jjerry_gdCollaborators
7 days ago Added label: bug-report
Jjerry_gdCollaborators
7 days ago Modify description of issue
Jjerry_gdCollaborators
7 days ago Modify description of issue
Jjerry_gdCollaborators
7 days ago Assign jerry_gd as responsible person
CANN-robotCANN-robotCollaborators
7 days ago Close issue
CANN-robotCANN-robotCollaborators
7 days ago Added label: resolved