| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912 | 25 天前 | |
【社区任务】aclnnRoll支持complex64 Co-authored-by: 繁依Fanyi<cyf1253238372@qq.com> # message auto-generated for no-merge-commit merge: !4250 merge aclnn-roll-complex64 into master 【社区任务】aclnnRoll支持complex64 Created-by: Twilight-Fanyi Commit-by: 繁依Fanyi Merged-by: cann-robot Description: ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopyPad 搬移路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的 Issue - [需求 Issue #2597：aclnnRoll 算子支持 complex64](https://gitcode.com/cann/ops-math/issues/2597) - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 ddfbc663：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；API/Host/Kernel UT、静态检查和性能数据已保存。 - A5 真机：ascend950 构建与安装通过；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50。 - 最终提交补充官方自测覆盖要求中的 BOOL 支持及回归用例，complex64 Kernel 搬移方案未变。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopy 路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的Issue - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 9f4b4612：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机（Kernel 等价版本 c3ca01a0）：功能/原 dtype 回归 21/21；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机（同一 Kernel 版本）：功能/原 dtype 回归 21/21。 - 最终提交相对上述真机版本只收窄 API 格式检查、恢复 Host 输入输出 shape 一致性并补测试，Kernel 未变；本次未将 WebIDE 终端不可用的启动过程计为复测通过。 - A5：ascend950 二进制构建通过；在 CANN 9.0.0 官方 CANN Simulator（DAV_3510）运行 aclnnRoll complex64 示例，bit-exact 检查通过。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4250 | 2 个月前 | |
【社区任务】aclnnRoll支持complex64 Co-authored-by: 繁依Fanyi<cyf1253238372@qq.com> # message auto-generated for no-merge-commit merge: !4250 merge aclnn-roll-complex64 into master 【社区任务】aclnnRoll支持complex64 Created-by: Twilight-Fanyi Commit-by: 繁依Fanyi Merged-by: cann-robot Description: ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopyPad 搬移路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的 Issue - [需求 Issue #2597：aclnnRoll 算子支持 complex64](https://gitcode.com/cann/ops-math/issues/2597) - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 ddfbc663：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；API/Host/Kernel UT、静态检查和性能数据已保存。 - A5 真机：ascend950 构建与安装通过；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50。 - 最终提交补充官方自测覆盖要求中的 BOOL 支持及回归用例，complex64 Kernel 搬移方案未变。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopy 路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的Issue - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 9f4b4612：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机（Kernel 等价版本 c3ca01a0）：功能/原 dtype 回归 21/21；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机（同一 Kernel 版本）：功能/原 dtype 回归 21/21。 - 最终提交相对上述真机版本只收窄 API 格式检查、恢复 Host 输入输出 shape 一致性并补测试，Kernel 未变；本次未将 WebIDE 终端不可用的启动过程计为复测通过。 - A5：ascend950 二进制构建通过；在 CANN 9.0.0 官方 CANN Simulator（DAV_3510）运行 aclnnRoll complex64 示例，bit-exact 检查通过。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4250 | 2 个月前 | |
【社区任务】aclnnRoll支持complex64 Co-authored-by: 繁依Fanyi<cyf1253238372@qq.com> # message auto-generated for no-merge-commit merge: !4250 merge aclnn-roll-complex64 into master 【社区任务】aclnnRoll支持complex64 Created-by: Twilight-Fanyi Commit-by: 繁依Fanyi Merged-by: cann-robot Description: ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopyPad 搬移路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的 Issue - [需求 Issue #2597：aclnnRoll 算子支持 complex64](https://gitcode.com/cann/ops-math/issues/2597) - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 ddfbc663：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；API/Host/Kernel UT、静态检查和性能数据已保存。 - A5 真机：ascend950 构建与安装通过；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50。 - 最终提交补充官方自测覆盖要求中的 BOOL 支持及回归用例，complex64 Kernel 搬移方案未变。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopy 路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的Issue - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 9f4b4612：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机（Kernel 等价版本 c3ca01a0）：功能/原 dtype 回归 21/21；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机（同一 Kernel 版本）：功能/原 dtype 回归 21/21。 - 最终提交相对上述真机版本只收窄 API 格式检查、恢复 Host 输入输出 shape 一致性并补测试，Kernel 未变；本次未将 WebIDE 终端不可用的启动过程计为复测通过。 - A5：ascend950 二进制构建通过；在 CANN 9.0.0 官方 CANN Simulator（DAV_3510）运行 aclnnRoll complex64 示例，bit-exact 检查通过。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4250 | 2 个月前 | |
【社区任务】aclnnRoll支持complex64 Co-authored-by: 繁依Fanyi<cyf1253238372@qq.com> # message auto-generated for no-merge-commit merge: !4250 merge aclnn-roll-complex64 into master 【社区任务】aclnnRoll支持complex64 Created-by: Twilight-Fanyi Commit-by: 繁依Fanyi Merged-by: cann-robot Description: ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopyPad 搬移路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的 Issue - [需求 Issue #2597：aclnnRoll 算子支持 complex64](https://gitcode.com/cann/ops-math/issues/2597) - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 ddfbc663：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；API/Host/Kernel UT、静态检查和性能数据已保存。 - A5 真机：ascend950 构建与安装通过；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50。 - 最终提交补充官方自测覆盖要求中的 BOOL 支持及回归用例，complex64 Kernel 搬移方案未变。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopy 路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的Issue - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 9f4b4612：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机（Kernel 等价版本 c3ca01a0）：功能/原 dtype 回归 21/21；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机（同一 Kernel 版本）：功能/原 dtype 回归 21/21。 - 最终提交相对上述真机版本只收窄 API 格式检查、恢复 Host 输入输出 shape 一致性并补测试，Kernel 未变；本次未将 WebIDE 终端不可用的启动过程计为复测通过。 - A5：ascend950 二进制构建通过；在 CANN 9.0.0 官方 CANN Simulator（DAV_3510）运行 aclnnRoll complex64 示例，bit-exact 检查通过。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4250 | 2 个月前 | |
【社区任务】aclnnRoll支持complex64 Co-authored-by: 繁依Fanyi<cyf1253238372@qq.com> # message auto-generated for no-merge-commit merge: !4250 merge aclnn-roll-complex64 into master 【社区任务】aclnnRoll支持complex64 Created-by: Twilight-Fanyi Commit-by: 繁依Fanyi Merged-by: cann-robot Description: ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopyPad 搬移路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的 Issue - [需求 Issue #2597：aclnnRoll 算子支持 complex64](https://gitcode.com/cann/ops-math/issues/2597) - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 ddfbc663：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机：功能与回归 22/22；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50；API/Host/Kernel UT、静态检查和性能数据已保存。 - A5 真机：ascend950 构建与安装通过；任务书 complex64 正式用例 100/100；原 dtype 正式用例 50/50。 - 最终提交补充官方自测覆盖要求中的 BOOL 支持及回归用例，complex64 Kernel 搬移方案未变。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## 描述 - 为 aclnnRoll 新增 complex64 输入/输出支持，保持接口签名和 ND 格式约束不变。 - Kernel 将 complex64 作为 8 字节整体映射到 uint64_t 搬运，复用现有 Roll 坐标映射和 DataCopy 路径，不参与复数运算。 - 使用平台 GetUbBlockSize 获取 Block 大小，并补齐非对齐/跨 UB 场景的安全处理。 - 增加 A2/A3/A5 配置、API/Host/Kernel UT、真机自测脚本、示例和文档。 ## 关联的Issue - [社区任务书：aclnnRoll 支持 complex64](https://gitcode.com/cann/cann-ops-competitions/blob/master/04_tasks/01_community-task-2026/docs/202607/aclnnRoll_task_doc.md) - [算子设计文档 PR #711](https://gitcode.com/cann/cann-ops-competitions/pull/711) ## 测试 - 最终提交 9f4b4612：clang-format 18.1.8、Ruff、Python 语法、JSON、diff-check、OAT、GitCode 服务端 hooks 均通过。 - A2 真机（Kernel 等价版本 c3ca01a0）：功能/原 dtype 回归 21/21；AscendOpTest 13/13；已保存原 dtype 性能对比结果。 - A3 真机（同一 Kernel 版本）：功能/原 dtype 回归 21/21。 - 最终提交相对上述真机版本只收窄 API 格式检查、恢复 Host 输入输出 shape 一致性并补测试，Kernel 未变；本次未将 WebIDE 终端不可用的启动过程计为复测通过。 - A5：ascend950 二进制构建通过；在 CANN 9.0.0 官方 CANN Simulator（DAV_3510）运行 aclnnRoll complex64 示例，bit-exact 检查通过。 ## 文档更新 - 更新 experimental/math/roll/README.md、docs/aclnnRoll.md。 - 新增 tests/selftest/README.md 和 complex64 真机自测脚本。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4250 | 2 个月前 | |
【社区任务】Add roll operator Co-authored-by: boxw987<boxw987@gmail.com> # message auto-generated for no-merge-commit merge: !3333 merge roll-submit-master into master 【社区任务】Add roll operator Created-by: boxw987 Commit-by: boxw987 Merged-by: cann-robot Description: ## 描述 本次提交补齐 Roll 算子的开源实现，面向 Atlas A2 训练系列产品提供 ACLNN 两阶段调用接口。 Roll 的目标是沿指定维度对输入张量执行循环位移；当 dims 为空时，语义等价于先对输入做逻辑展平，再执行一维 roll，最后恢复原始形状。 本次实现重点解决以下问题： 1. 正式支持范围内的语义对齐。 2. 单维、多维、最后一维、非最后一维等不同访存形态下的统一实现。 3. 在控制实现复杂度的前提下，保证正式支持范围内的性能收益。 当前版本支持范围如下： - 数据类型：uint8、int8、bfloat16、float16、float32、int32、uint32 - 数据格式：ND - 维度范围：0-D 到 8-D - 语义特性：负维度归一化、重复维度合并、dims 为空的 flatten roll 当前版本不支持： - bool - int64 - 非 ND 格式 实现上采用 Host 侧归一化与 Tiling、Kernel 侧多路径搬运的方案： mermaid flowchart TD A[ACLNN 接口] --> B[参数校验] B --> C[dims shifts 归一化] C --> D[重复维度合并] D --> E[生成 Tiling 信息] E --> F[Kernel 路径选择] F --> G[连续段 Copy] F --> H[按行 Roll] F --> I[按 Block Roll] F --> J[Segmented Fallback] 整体设计思路如下： 1. Host 侧负责参数校验、负维归一化、重复维度合并和切分信息生成。 2. dims 为空时退化为 flatten roll，避免继续走原始多维索引路径。 3. 单 active dim 与多 active dim 分开建模，分别提取更适合搬运的结构信息。 4. 最后一维滚动优先按行处理，非最后维滚动优先按 block 重排。 5. 保留 segmented fallback 作为通用正确性兜底路径。 Kernel 路径选择可概括为： mermaid flowchart TD A[Kernel Process] --> B{activeDimCount == 0?} B -->|Yes| C[Identity Copy] B -->|No| D{dimNum == 1?} D -->|Yes| E[Flatten Roll] D -->|No| F{single active dim?} F -->|Yes| G{activeDim == 0?} G -->|Yes| H[Leading Dim Roll] G -->|No| I{innerSize == 1 and last dim?} I -->|Yes| J[Last Dim Row Roll] I -->|No| K[Single Dim Block Roll] F -->|No| L{last active dim is final dim?} L -->|Yes| M[Multi-Dim Last-Dim Roll] L -->|No| N[Multi-Dim Non-Last Roll] N --> O[Segmented Fallback] 本次实现保留的优化主线不是单点 shape 特化，而是以下几类可复用策略： 1. 连续段优先 能转为连续段搬运的场景，优先使用连续 copy 或源端对齐 copy，减少逐元素处理。 2. 最后一维按行优化 围绕最后一维滚动构建按行搬运、按行拼接和行组处理路径，降低访存碎片。 3. 非最后维按 block 重排 对中间维或多维组合滚动，优先按 block 建模，减少重复索引计算。 4. 小宽度定向增强 对部分 BF16、小宽度最后维和部分 uint8 小块场景，调整切分与 patch 策略，但不引入写死单一 shape 的独立实现。 输入约束方面，当前版本要求： 1. 输入输出 dtype 一致。 2. 输入输出 shape 一致。 3. rank 不超过 8。 4. dims 为空时，shifts 长度必须为 1。 5. dims 非空时，shifts 与 dims 长度必须一致。 6. dims 的取值范围为 [-rank, rank)。 7. 0 维输入时，shifts 长度必须为 1，且 dims 为空。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/1971 ## 测试 本次改动包含以下验证范围： 1. ACLNN 接口参数校验用例 2. Host 侧 InferShape / Tiling 用例 3. Kernel 路径基础功能用例 4. dims 为空、负维度、重复维度合并等关键语义场景 5. 正式支持 dtype 范围内的基础调用样例 验收目标包括： 1. 正式支持范围内功能结果正确。 2. 单维、多维、负维和重复维度归并场景行为正确。 3. 正式支持范围内的性能目标优于参考 TBE。 ## 文档更新 本次同步补充了算子说明文档、接口文档以及需求与设计说明。 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [x] 文档更新 - [x] 其他，请描述：新增 Roll 算子实现与配套文档 See merge request: cann/ops-math!3333 | 3 个月前 | |
产品名变更 Co-authored-by: gitcode-chenjiao<chenjiao31@huawei.com> # message auto-generated for no-merge-commit merge: !5912 merge master into master 产品名变更 Created-by: gitcode-chenjiao Commit-by: gitcode-chenjiao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> AI处理器对应的产品型号名称更新 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#3258](https://gitcode.com/cann/ops-math/issues/3258) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ok ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5912 | 25 天前 |
Roll
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| boxw987 | 个人开发者 | Roll | 2026/06 | Roll 算子适配开源仓 |
支持的产品型号
- Atlas A2训练系列产品
- Atlas A3训练系列产品
- Ascend 950PR&950DT系列产品
算子描述
-
功能描述
Roll沿给定维度对输入张量执行循环位移；当dims为空时，先按逻辑视图展平，再执行一维循环位移，最后按原始形状输出。 -
原型信息
算子类型(OpType) Roll name Type data type format 算子输入 x tensor bool, uint8, int8, bfloat16, float16, float32, int32, uint32, complex64 ND 算子输出 y tensor 与 x 相同 ND 属性 shifts listInt 整型列表 - dims listInt 整型列表 - 核函数名 roll
约束与限制
- 仅支持
ND格式。 - 支持 0 维到 8 维输入。
dims为空时，shifts长度必须为 1。dims非空时，shifts与dims长度必须一致。dims取值范围为[-rank, rank)。
算子使用
使用该算子前，请参考社区版 CANN 开发套件包安装文档完成开发运行环境部署。
编译部署
cd ${git_clone_path}/ops-math
bash build.sh --pkg --experimental --soc=ascend910b --ops=roll
./build_out/cann-ops-<vendor_name>-linux.<arch>.run
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn 调用 | test_aclnn_roll.cpp | 通过 aclnnRoll 接口方式调用 Roll 算子 |