| 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!1633 | 2 个月前 |