0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robotadd dynamic_stitch infershape
3a691133创建于 2 天前历史提交
文件最后提交记录最后更新时间
examples
feat: 新增 DynamicStitch 算子实现 Co-authored-by: 王荀<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !1633 merge dynamic_stitch_fix into master feat: 新增 DynamicStitch 算子实现 Created-by: wangxun21 Commit-by: wangxun21;王荀 Merged-by: cann-robot Description: ## 描述 本 PR 新增了 DynamicStitch 算子的完整实现，该算子与 TensorFlow 的 DynamicStitch 算子兼容。 DynamicStitch 算子用于将多个张量的数据交织到单个张量中，支持多种数据类型。 ### 改动原因 支持 TensorFlow 模型中 DynamicStitch 算子的迁移，丰富 ops-math 仓库的算子支持范围。 ### 改动方法 1. 新增算子原型定义（op_graph/dynamic_stitch_proto.h） 2. 新增 host 侧实现： - 算子属性配置（op_host/dynamic_stitch_def.cpp） - Tiling 配置（op_host/config/ascend950/） 3. 新增 kernel 侧实现： - 通用头文件（op_kernel/arch35/dynamic_stitch_common.h） - Indices 去重（op_kernel/arch35/dynamic_stitch_indices_deduplicate.h） - SIMD/SIMT 实现（op_kernel/arch35/） 4. 新增单元测试（tests/ut/） 5. 新增示例代码（examples/test_geir_dynamic_stitch.cpp） ## 关联的Issue - #998 ## 测试 - 已添加单元测试 test_dynamic_stitch_tiling.cpp - 已添加示例代码 test_geir_dynamic_stitch.cpp ## 文档更新 无 ## 类型标签 - [x] 新特性 See merge request: cann/ops-math!16332 个月前
op_graph
feat: 新增 DynamicStitch 算子实现 Co-authored-by: 王荀<wangxun21@huawei.com> # message auto-generated for no-merge-commit merge: !1633 merge dynamic_stitch_fix into master feat: 新增 DynamicStitch 算子实现 Created-by: wangxun21 Commit-by: wangxun21;王荀 Merged-by: cann-robot Description: ## 描述 本 PR 新增了 DynamicStitch 算子的完整实现，该算子与 TensorFlow 的 DynamicStitch 算子兼容。 DynamicStitch 算子用于将多个张量的数据交织到单个张量中，支持多种数据类型。 ### 改动原因 支持 TensorFlow 模型中 DynamicStitch 算子的迁移，丰富 ops-math 仓库的算子支持范围。 ### 改动方法 1. 新增算子原型定义（op_graph/dynamic_stitch_proto.h） 2. 新增 host 侧实现： - 算子属性配置（op_host/dynamic_stitch_def.cpp） - Tiling 配置（op_host/config/ascend950/） 3. 新增 kernel 侧实现： - 通用头文件（op_kernel/arch35/dynamic_stitch_common.h） - Indices 去重（op_kernel/arch35/dynamic_stitch_indices_deduplicate.h） - SIMD/SIMT 实现（op_kernel/arch35/） 4. 新增单元测试（tests/ut/） 5. 新增示例代码（examples/test_geir_dynamic_stitch.cpp） ## 关联的Issue - #998 ## 测试 - 已添加单元测试 test_dynamic_stitch_tiling.cpp - 已添加示例代码 test_geir_dynamic_stitch.cpp ## 文档更新 无 ## 类型标签 - [x] 新特性 See merge request: cann/ops-math!16332 个月前
op_host
add dynamic_stitch infershape Co-authored-by: alfengyuan<2640706467@qq.com> # message auto-generated for no-merge-commit merge: !3063 merge master into master add dynamic_stitch infershape Created-by: alfengyuan Commit-by: alfengyuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> DynamicStitch 算子补充infershape交付件 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1761 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 冒烟通过，算子泛化精度性能pass， 图模式测试pass ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!30632 天前
op_kernel
[kernel] dynamic_stitch histogram_v2 and one_hot modify simt api Co-authored-by: majiayuan<majiayuan1@huawei.com> # message auto-generated for no-merge-commit merge: !2483 merge master into master [kernel] dynamic_stitch histogram_v2 and one_hot modify simt api Created-by: majiayuan Commit-by: majiayuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 根据映射关系，统一将C++风格API替换为C风格API。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [Requirement|需求建议]: dynamic_stitch, histogram_v2 以及 one_hot SIMT算子整改原有cpp simt接口 [#1492](https://gitcode.com/cann/ops-math/issues/1492) ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> SIMT API接口修改前后，反汇编得到的md5码相同。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!248324 天前
tests
add dynamic_stitch infershape Co-authored-by: alfengyuan<2640706467@qq.com> # message auto-generated for no-merge-commit merge: !3063 merge master into master add dynamic_stitch infershape Created-by: alfengyuan Commit-by: alfengyuan Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> DynamicStitch 算子补充infershape交付件 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1761 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 冒烟通过，算子泛化精度性能pass， 图模式测试pass ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 不涉及 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!30632 天前
CMakeLists.txt
fix dynamicStitch Co-authored-by: xieshengwei1024<xieshengwei3@huawei.com> # message auto-generated for no-merge-commit merge: !1368 merge master into master fix dynamicStitch Created-by: xieshengwei1024 Commit-by: xieshengwei1024 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 解决dynamicStitch aicore不使能的问题 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [x] 其他，请描述：ut增补 See merge request: cann/ops-math!13682 个月前