已合并
feat: 新增 DynamicStitch 算子实现 #1633
wangxun21创建于 3月12日
feat: 新增 DynamicStitch 算子实现 #1633
已合并
wangxun21创建于 3月12日
wangxun21成员
3月12日

描述

本 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

文档更新

类型标签

  • 新特性
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 wangxun21 的贡献)
Wwangxun21成员
3月12日 创建了 pull request,commit 525c6404
CANN-robot
CANN-robot成员
3月12日 评论:

🔵 source code change are detected, tasks labels is removed in this pull request!

likedislike
CANN-robotCANN-robot成员
3月12日 添加了label:stat/needs-squash
CANN-robot
CANN-robot成员
3月12日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
*/*/op_graph/*_proto.h tang-lei01, songkai111, zhou-qilong (3/2) tang-lei01 (1/1)
*/*/op_host/*_def.cpp tang-lei01, zhou-qilong, songkai111 (3/2) tang-lei01 (1/1)
repo-cann/ops-math zhou-qilong, songkai111 (2/2) zhou-qilong, songkai111 (2/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

wangxun21, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
CANN-robotCANN-robot成员
3月12日 将zhou-qilong,wangrui_,rxtfeng,loov1,songkai111,llimwang,gubaocheng设为评审人
此处折叠了103条消息 查看更多
sunday成员
3月18日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
3月18日 添加了label:lgtm
CANN-robot
CANN-robot成员
3月18日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: zhou-qilong, tang-lei01, songkai111.
Reviewers who wrote a comment of /lgtm are: tang-lei01, zhou-qilong, songkai111.

likedislike
CANN-robotCANN-robot成员
3月18日 关闭了关联的issue
CANN-robotCANN-robot成员
3月18日 合入了pull request