描述

为 diag、arg_max_v2、arg_max_with_value、arg_min、arg_min_with_value 五个算子新增 ascend960dt 芯片平台适配，使其可在 ascend960dt（arch35）上完成算子定义注册、tiling 编译与二进制 kernel 配置生成。

改动原因

算子 v121 分支合入主线，并为上述 5 个算子补齐 ascend960dt 平台支持。

改动方法

1. 各算子 CMakeLists.txt 的 SUPPORT_COMPUTE_UNIT 新增 ascend960dt，SUPPORT_TILING_DIR 对应追加 arch35，复用 arch35 tiling 实现。
2. 各算子 op_host/*_def.cpp 增加 this->AICore().AddConfig("ascend960dt", aicore_config)，完成 AICore 配置注册。
3. 新增 op_host/config/ascend960dt/ 目录，包含 {op}_binary.json（二进制 kernel 配置）与 {op}_simplified_key.ini（simplified_key_mode 配置，default=0）。
4. scripts/kernel/binary_config/ascendc_config.json 为 Diag/ArgMaxWithValue/ArgMaxV2/ArgMinWithValue/ArgMin 增加 ascend960dt compute_units，并配置编译选项 -mllvm -cce-aicore-dcci-before-kernel-end=false。
5. math/arg_min/framework/argmin_onnx_plugin.cpp 的 IsRegBasePlatform() 将 NpuArch::DAV_9201 加入 regbase 架构集合。

变更文件

• conversion/diag：CMakeLists.txt、op_host/diag_def.cpp、新增 config/ascend960dt/{diag_binary.json, diag_simplified_key.ini}
• math/arg_max_v2：CMakeLists.txt、op_host/arg_max_v2_def.cpp、新增 config/ascend960dt/{arg_max_v2_binary.json, arg_max_v2_simplified_key.ini}
• math/arg_max_with_value：CMakeLists.txt、op_host/arg_max_with_value_def.cpp、新增 config/ascend960dt/{arg_max_with_value_binary.json, arg_max_with_value_simplified_key.ini}
• math/arg_min：CMakeLists.txt、framework/argmin_onnx_plugin.cpp、op_host/arg_min_def.cpp、新增 config/ascend960dt/{arg_min_binary.json, arg_min_simplified_key.ini}
• math/arg_min_with_value：CMakeLists.txt、op_host/arg_min_with_value_def.cpp、新增 config/ascend960dt/{arg_min_with_value_binary.json, arg_min_with_value_simplified_key.ini}
• scripts/kernel/binary_config/ascendc_config.json

关联的Issue

• 关联 Issue：https://gitcode.com/cann/ops-math/issues/3306

测试

本次改动为芯片平台适配（CMake 编译配置、算子定义注册、binary/simplified_key 配置），未修改 kernel 计算逻辑，复用 arch35 tiling 与 kernel 实现。需在 ascend960dt 环境完成编译及算子泛化、二级冒烟验证。

文档更新

无

类型标签

• Bug修复
• 新特性
• 性能优化
• 文档更新
• 其他，请描述：