master

cann-robotFeat: 新增 aclsparseSgpsvInterleavedBatch 五对角批量求解接口（Ascend950/arch35）

66eeda4d创建于 6 天前历史提交

文件	最后提交记录	最后更新时间
csv_loader.h	feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !46 merge aclsparseSgtsvInterleavedBatch into master feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 为 Ascend 950PR / Ascend 950DT（arch35）新增 aclsparseSgtsvInterleavedBatch 算子实现，支持批量求解多个独立的三对角线性方程组 **A^(k) · x^(k) = b^(k)**。 改动原因： - ops-sparse 库此前缺少三对角线性方程组直接求解器，无法覆盖科学计算 / PDE 离散化 / 隐式时间积分等场景的高频需求 - 需对齐 NVIDIA cuSPARSE cusparseSgtsvInterleavedBatch 的接口语义，补齐 CANN Ascend 950 系列稀疏计算能力 改动方法： - 新增对外接口 aclsparseSgtsvInterleavedBatch（主接口）与 aclsparseSgtsvInterleavedBatch_bufferSizeExt（workspace 查询），定义于 include/cann_ops_sparse.h - 基于 SIMT 模型实现 kernel（src/gtsv_interleaved_batch/arch35/）：每 block 64 线程，每线程独立处理 1 个 batch；线程局部状态保持在寄存器中，跨行依赖不经过 workspace GM 读写 - 采用 Thomas 算法（algo=0），workspace 仅存 d'[i] 供 backward 阶段使用，b'[i] 原位覆写到 x - 重构测试框架（新增 test/frame/ 公共组件：csv_loader、test_common、test_main），适配 CSV 数据驱动的算子测试 - 新增完整算子文档 src/gtsv_interleaved_batch/README.md ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> - #88 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 新增 test/gtsv_interleaved_batch/ 测试用例，基于 CSV 数据驱动（gtsv_interleaved_batch_test.csv），覆盖 m=1 / 多 batch / 边界值等场景 - 提供 golden 参考实现（gtsv_interleaved_batch_golden.h）对比 NPU 计算结果 - 在 Ascend 950 上验证正确性（详见 README 预期输出样例）  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 新增 src/gtsv_interleaved_batch/README.md：包含算子概述、接口函数原型、参数说明、约束条件、调用示例及预期输出 - 更新 include/cann_ops_sparse.h：新增 aclsparseSgtsvInterleavedBatch 与 _bufferSizeExt 接口声明及 Doxygen 注释 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!46	9 天前
descriptor_manager.h	feat: add aclsparseSnnz operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !34 merge aclsparseSnnz into master feat: add aclsparseSnnz operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1、A5上支持nnz算子 2、新增测试框架，不影响已上库算子 3、引入workspace 4、引入host、device模式的set、get接口 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #77 > Issue 链接：https://gitcode.com/cann/ops-sparse/issues/77 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> nnz的29个用例全部执行通过  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!34	16 天前
fill.h	Feat: 新增 aclsparseSgpsvInterleavedBatch 五对角批量求解接口（Ascend950/arch35） Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !55 merge aclsparseSgpsvInterleavedBatch into master Feat: 新增 aclsparseSgpsvInterleavedBatch 五对角批量求解接口（Ascend950/arch35） Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 新增 aclsparseSgpsvInterleavedBatch 接口，实现面向 Ascend950（arch35）的五对角批次线性方程组求解。 **算子功能**：批量求解五对角线性方程组 A^(k) · x^(k) = b^(k)，其中 A^(k) 为 m×m 五对角矩阵，由 5 条对角线（ds, dl, d, du, dw）定义，数据布局为 row-major interleaved。 **目标芯片**：Ascend950（A5 架构，arch35） **编程模型**：SIMT — 每线程处理 1 个 batch 的完整 m 行扫描，batch 间无数据依赖，通过 grid-stride 支持 batchCount 超过硬件核心数的场景。 **算法**：QR 分解 **接口列表**： - aclsparseSgpsvInterleavedBatch — 求解接口（FP32） - aclsparseSgpsvInterleavedBatch_bufferSizeExt — 工作区大小查询接口 ## 关联 Issue: <!-- Issue 提交后补充链接，格式：[#N](https://gitcode.com/cann/ops-sparse/issues/N) --> https://gitcode.com/cann/ops-sparse/issues/96 ## 测试 **测试框架**：GTest + CSV 参数化 **测试用例**：共 54 条，全部通过 - 正向用例：35条（L0 12 条 + L1 24 条） - 异常用例：19 条（nullptr / 非法参数 / 对齐错误等）  ## 文档更新 - **新增** src/gpsv_interleaved_batch/README.md — 算子接口文档（概述、接口签名、参数说明、约束、调用示例、LU 算法附录） - **修改** include/cann_ops_sparse.h — 追加 gpsv section 接口声明 - **新增** test/frame/fill.h — 追加约 100 行 Pentadiag 矩阵生成器（五对角填充 helper） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!55	6 天前
sparse_test.h	feat: add aclsparseSnnz operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !34 merge aclsparseSnnz into master feat: add aclsparseSnnz operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1、A5上支持nnz算子 2、新增测试框架，不影响已上库算子 3、引入workspace 4、引入host、device模式的set、get接口 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #77 > Issue 链接：https://gitcode.com/cann/ops-sparse/issues/77 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> nnz的29个用例全部执行通过  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!34	16 天前
test_common.h	feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !46 merge aclsparseSgtsvInterleavedBatch into master feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 为 Ascend 950PR / Ascend 950DT（arch35）新增 aclsparseSgtsvInterleavedBatch 算子实现，支持批量求解多个独立的三对角线性方程组 **A^(k) · x^(k) = b^(k)**。 改动原因： - ops-sparse 库此前缺少三对角线性方程组直接求解器，无法覆盖科学计算 / PDE 离散化 / 隐式时间积分等场景的高频需求 - 需对齐 NVIDIA cuSPARSE cusparseSgtsvInterleavedBatch 的接口语义，补齐 CANN Ascend 950 系列稀疏计算能力 改动方法： - 新增对外接口 aclsparseSgtsvInterleavedBatch（主接口）与 aclsparseSgtsvInterleavedBatch_bufferSizeExt（workspace 查询），定义于 include/cann_ops_sparse.h - 基于 SIMT 模型实现 kernel（src/gtsv_interleaved_batch/arch35/）：每 block 64 线程，每线程独立处理 1 个 batch；线程局部状态保持在寄存器中，跨行依赖不经过 workspace GM 读写 - 采用 Thomas 算法（algo=0），workspace 仅存 d'[i] 供 backward 阶段使用，b'[i] 原位覆写到 x - 重构测试框架（新增 test/frame/ 公共组件：csv_loader、test_common、test_main），适配 CSV 数据驱动的算子测试 - 新增完整算子文档 src/gtsv_interleaved_batch/README.md ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> - #88 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 新增 test/gtsv_interleaved_batch/ 测试用例，基于 CSV 数据驱动（gtsv_interleaved_batch_test.csv），覆盖 m=1 / 多 batch / 边界值等场景 - 提供 golden 参考实现（gtsv_interleaved_batch_golden.h）对比 NPU 计算结果 - 在 Ascend 950 上验证正确性（详见 README 预期输出样例）  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 新增 src/gtsv_interleaved_batch/README.md：包含算子概述、接口函数原型、参数说明、约束条件、调用示例及预期输出 - 更新 include/cann_ops_sparse.h：新增 aclsparseSgtsvInterleavedBatch 与 _bufferSizeExt 接口声明及 Doxygen 注释 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!46	9 天前
test_main.cpp	feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !46 merge aclsparseSgtsvInterleavedBatch into master feat: add aclsparseSgtsvInterleavedBatch operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 为 Ascend 950PR / Ascend 950DT（arch35）新增 aclsparseSgtsvInterleavedBatch 算子实现，支持批量求解多个独立的三对角线性方程组 **A^(k) · x^(k) = b^(k)**。 改动原因： - ops-sparse 库此前缺少三对角线性方程组直接求解器，无法覆盖科学计算 / PDE 离散化 / 隐式时间积分等场景的高频需求 - 需对齐 NVIDIA cuSPARSE cusparseSgtsvInterleavedBatch 的接口语义，补齐 CANN Ascend 950 系列稀疏计算能力 改动方法： - 新增对外接口 aclsparseSgtsvInterleavedBatch（主接口）与 aclsparseSgtsvInterleavedBatch_bufferSizeExt（workspace 查询），定义于 include/cann_ops_sparse.h - 基于 SIMT 模型实现 kernel（src/gtsv_interleaved_batch/arch35/）：每 block 64 线程，每线程独立处理 1 个 batch；线程局部状态保持在寄存器中，跨行依赖不经过 workspace GM 读写 - 采用 Thomas 算法（algo=0），workspace 仅存 d'[i] 供 backward 阶段使用，b'[i] 原位覆写到 x - 重构测试框架（新增 test/frame/ 公共组件：csv_loader、test_common、test_main），适配 CSV 数据驱动的算子测试 - 新增完整算子文档 src/gtsv_interleaved_batch/README.md ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> - #88 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 新增 test/gtsv_interleaved_batch/ 测试用例，基于 CSV 数据驱动（gtsv_interleaved_batch_test.csv），覆盖 m=1 / 多 batch / 边界值等场景 - 提供 golden 参考实现（gtsv_interleaved_batch_golden.h）对比 NPU 计算结果 - 在 Ascend 950 上验证正确性（详见 README 预期输出样例）  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 新增 src/gtsv_interleaved_batch/README.md：包含算子概述、接口函数原型、参数说明、约束条件、调用示例及预期输出 - 更新 include/cann_ops_sparse.h：新增 aclsparseSgtsvInterleavedBatch 与 _bufferSizeExt 接口声明及 Doxygen 注释 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!46	9 天前
types.h	feat: add aclsparseSnnz operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !34 merge aclsparseSnnz into master feat: add aclsparseSnnz operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1、A5上支持nnz算子 2、新增测试框架，不影响已上库算子 3、引入workspace 4、引入host、device模式的set、get接口 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #77 > Issue 链接：https://gitcode.com/cann/ops-sparse/issues/77 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> nnz的29个用例全部执行通过  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!34	16 天前
verify.h	feat: add aclsparseSnnz operator for Ascend950 (arch35) Co-authored-by: justsheldon<taoxudong@huawei.com> # message auto-generated for no-merge-commit merge: !34 merge aclsparseSnnz into master feat: add aclsparseSnnz operator for Ascend950 (arch35) Created-by: justsheldon Commit-by: justsheldon Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1、A5上支持nnz算子 2、新增测试框架，不影响已上库算子 3、引入workspace 4、引入host、device模式的set、get接口 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> #77 > Issue 链接：https://gitcode.com/cann/ops-sparse/issues/77 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> nnz的29个用例全部执行通过  ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-sparse!34	16 天前