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,需要转化。
一、问题描述 (必填)
ascend310B TopKFusionPass融合规则执行后返回 data_type DT_INT64 of input[1, k] is unsupported
二、环境信息 (可选)
ascend310B
三、重现步骤 (可选)
四、预期结果 (可选)
ONNX 规范中 TopK 的 k 是 int64,TopKFusionPass处理ascend310B的TopK节点时,aicpu TopK不支持int64,需要转化。
💡 备注(选填)