Pull Request已成功合入, 合并人@CANN-robot
（感谢 sunchun 的贡献）
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| math/floor_div | ✅ rxtfeng, FelixTang7 (2/2) | ✅ rxtfeng (1/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
CANN-robotCANN-robot🔵 source code change are detected, tasks labels is removed in this pull request!
问题/功能描述
本次PR旨在扩展FloorDiv算子在DAV_2201 NPU架构上的AiCore数据类型支持。主要解决了该算子在新硬件平台上无法利用AiCore高性能计算单元执行的问题，确保了算子在不同NPU架构下功能与性能的一致性。
修改方案描述
修改了GetAiCoreDtypeSupportListBySocVersion函数中的switch-case逻辑，将NpuArch::DAV_2201添加到与NpuArch::DAV_3510相同的处理分支中。这使得当运行在DAV_2201架构上时，FloorDiv算子能返回包含DT_INT64等更多数据类型的支持列表，从而能够调用AiCore内核进行计算，提升了算子的执行效率。
流水线任务触发成功
任务链接 [a67a52516e3b4b1e82e5601770a6f7a7][流水线指导]
| 任务名称 | 状态 | 日志 | 下载链接 |
|---|---|---|---|
| codecheck | ✅ SUCCESS | >>>>> | |
| SCA | ✅ SUCCESS | >>>>> | |
| anti_virus | ✅ SUCCESS | >>>>> | |
| Check_Pr | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_X86 | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_X86_mobile_station | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_ARM | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_X86_experimental | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_ARM_experimental | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_single | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_X86_A5 | ✅ SUCCESS | >>>>> | >>>>> |
| Compile_Ascend_ARM_A5 | ✅ SUCCESS | >>>>> | >>>>> |
| API_Check | ✅ SUCCESS | >>>>> | |
| pre_comment | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_X86_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_X86_mobile_station_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_X86_experimental_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_X86_A5_ubuntu24 | ✅ SUCCESS | >>>>> | |
| harmony-infer | ✅ SUCCESS | ||
| Compile_Ascend_ARM_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_ARM_experimental_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_single_ubuntu24 | ✅ SUCCESS | >>>>> | |
| Compile_Ascend_ARM_A5_ubuntu24 | ✅ SUCCESS | >>>>> | |
| UT_Test | ✅ SUCCESS | ||
| UT_Test_experimental | ✅ SUCCESS | ||
| UT_Test_kernel | ✅ SUCCESS | ||
| Smoke_A900_npupool | ✅ SUCCESS | >>>>> |
[2026-03-12 15:06:07] CI执行结束
/lgtm
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: rxtfeng.
Reviewers who wrote a comment of /lgtm are: rxtfeng, FelixTang7.
描述
floordiv A3上AICORE支持int64数据类型
关联的Issue
https://gitcode.com/cann/ops-math/issues/925
测试
文档更新
类型标签