Closed
[Bug-Report|缺陷反馈]: ascend960dt/ascend960pr平台算子二进制配置缺失,ConcatD融合pass未覆盖DAV_9201 #3681
yangjinwenAdd on  21 days agoClosed at  21 days ago
yangjinwenCollaborators
21 days ago Create

一、问题描述 (必填)

ascend960dt/ascend960pr 平台上存在以下问题:

  1. Concat→ConcatD 融合 pass 未生效:conversion/concat_d/op_graph/fusion_pass/concat_to_concatd_fusion_pass.cpp 中 IsTargetPlatform() 的 RegBase 架构白名单仅包含 DAV_3510/DAV_5102,未包含 ascend960dt 对应的 DAV_9201,导致该融合 pass 在 ascend960dt 平台整体跳过(日志输出 "Platform NpuArch xx is not regbase(DAV_3510/DAV_5102), skip whole graph.")。
  2. 算子二进制配置缺失:scripts/kernel/binary_config/ascendc_config.json 中 Concat、ConcatD、ConcatV2、Pack、BatchToSpaceND、MatrixSetDiag、SpaceToDepth、DepthToSpace、Rint 共 9 个算子缺失 ascend960dt 计算单元(compute_units)及对应编译选项配置,相关算子无法在 ascend960dt 平台编译运行。
  3. MatrixSetDiag 不支持 ascend960dt/ascend960pr:conversion/matrix_set_diag 工程的 CMakeLists(COMPUTE_UNIT/TILING_DIR)、matrix_set_diag_def.cpp(AICore 配置注册)及 op_host/config/ 下两平台的 matrix_set_diag_binary.json(int8/int16/int32/int64)均未覆盖。

二、环境信息 (可选)

• 昇腾硬件型号:ascend960dt / ascend960pr
• 仓库:cann/ops-math

三、重现步骤 (可选)

在 ascend960dt/ascend960pr 平台加载包含 Concat/ConcatD 节点的计算图,观察融合 pass 日志跳过整个图;同时上述算子在 ascend960dt 上无法编译生成二进制,MatrixSetDiag 无法调用。

四、预期结果 (可选)

  1. Concat→ConcatD 融合 pass 在 ascend960dt(DAV_9201)平台正常生效;
  2. 上述 9 个算子具备 ascend960dt 二进制编译配置,可正常编译运行;
  3. MatrixSetDiag 支持 ascend960dt/ascend960pr 平台(含 int8/int16/int32/int64 dtype 二进制)。

关联 PR:https://gitcode.com/cann/ops-math/merge_requests/6523

likedislike