master

cann-robot西工大智能感知交互实验室-AscendC实现AsStrided算子贡献

d2ec03c3Add on Jul 9History

Files	Last commit	Last update
as_strided.cpp	西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Co-authored-by: Mars_Cheng_cys<1214886529@qq.com> # message auto-generated for no-merge-commit merge: !3916 merge as-strided-experimental-20260708 into master 西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Created-by: Mars_Cheng_cys Commit-by: Mars_Cheng_cys Merged-by: cann-robot Description: ## 描述 使用 AscendC 对 TBE 实现的 AsStrided 算子进行 AICore 实现重构，实现 AscendC 版本的 AsStrided 算子对 Atlas A2 训练系列产品的适配。AsStrided 根据 size、stride 和 storage_offset 从输入 Tensor 的底层存储中按步长取数，生成指定 shape 的输出 Tensor，语义对齐 torch.as_strided 的取数规则。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2180 ## 测试 采用 aclnn 调用测试；当前提交前已完成 git diff --cached --check，Ascend 环境功能和精度建议以仓库 CI 或 ST 用例复核。 ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3916	3 months ago
as_strided.h	西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Co-authored-by: Mars_Cheng_cys<1214886529@qq.com> # message auto-generated for no-merge-commit merge: !3916 merge as-strided-experimental-20260708 into master 西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Created-by: Mars_Cheng_cys Commit-by: Mars_Cheng_cys Merged-by: cann-robot Description: ## 描述 使用 AscendC 对 TBE 实现的 AsStrided 算子进行 AICore 实现重构，实现 AscendC 版本的 AsStrided 算子对 Atlas A2 训练系列产品的适配。AsStrided 根据 size、stride 和 storage_offset 从输入 Tensor 的底层存储中按步长取数，生成指定 shape 的输出 Tensor，语义对齐 torch.as_strided 的取数规则。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2180 ## 测试 采用 aclnn 调用测试；当前提交前已完成 git diff --cached --check，Ascend 环境功能和精度建议以仓库 CI 或 ST 用例复核。 ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3916	3 months ago
as_strided_tiling_data.h	西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Co-authored-by: Mars_Cheng_cys<1214886529@qq.com> # message auto-generated for no-merge-commit merge: !3916 merge as-strided-experimental-20260708 into master 西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Created-by: Mars_Cheng_cys Commit-by: Mars_Cheng_cys Merged-by: cann-robot Description: ## 描述 使用 AscendC 对 TBE 实现的 AsStrided 算子进行 AICore 实现重构，实现 AscendC 版本的 AsStrided 算子对 Atlas A2 训练系列产品的适配。AsStrided 根据 size、stride 和 storage_offset 从输入 Tensor 的底层存储中按步长取数，生成指定 shape 的输出 Tensor，语义对齐 torch.as_strided 的取数规则。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2180 ## 测试 采用 aclnn 调用测试；当前提交前已完成 git diff --cached --check，Ascend 环境功能和精度建议以仓库 CI 或 ST 用例复核。 ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3916	3 months ago
as_strided_tiling_key.h	西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Co-authored-by: Mars_Cheng_cys<1214886529@qq.com> # message auto-generated for no-merge-commit merge: !3916 merge as-strided-experimental-20260708 into master 西工大智能感知交互实验室-AscendC实现AsStrided算子贡献 Created-by: Mars_Cheng_cys Commit-by: Mars_Cheng_cys Merged-by: cann-robot Description: ## 描述 使用 AscendC 对 TBE 实现的 AsStrided 算子进行 AICore 实现重构，实现 AscendC 版本的 AsStrided 算子对 Atlas A2 训练系列产品的适配。AsStrided 根据 size、stride 和 storage_offset 从输入 Tensor 的底层存储中按步长取数，生成指定 shape 的输出 Tensor，语义对齐 torch.as_strided 的取数规则。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2180 ## 测试 采用 aclnn 调用测试；当前提交前已完成 git diff --cached --check，Ascend 环境功能和精度建议以仓库 CI 或 ST 用例复核。 ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3916	3 months ago