已合并
MemSet算子开发-算子提交 #939
huangyuqian创建于 1月27日
MemSet算子开发-算子提交 #939
已合并
huangyuqian创建于 1月27日
已删除 :master合入到cann/ops-mathmaster
huangyuqian
1月27日

描述

开发背景

继承MemSet Tbe版本算子原型,开发Ascend C版本优化性能

功能支持

1、参考 - 昇腾社区-CANN商用版-Ascend C算子开发接口-Host API-原型注册与管理-OpParamDef-InitValue 所述支持范围,提供如下类型支持范围image.png
2、参考Tbe源码中image.png最大支持tensor数,Ascend C版本最大支持MemSet-tensor数192个
3、aclnn支持initvalue类型与outputTensor类型不匹配,Geir要求类型的字节大小匹配

tbe参数组装分析

1、算子原型如下:memset无输入输出tensor,依赖于框架组装传递参数
image.png
2、tbe通过识别动静态编译,从不同位置读取参数信息,该方法不再应用于ascendc,改为全采用TilingContext进行读取,不再区分动静态
image.png

关联的Issue

测试

文档更新

类型标签

  • Bug修复
  • 新特性
  • 性能优化
  • 文档更新
  • 其他,请描述:
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 huangyuqian 的贡献)
CANN-robot
CANN-robot成员
1月27日 评论:

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

likedislike
CANN-robot
CANN-robot成员
1月27日 评论:

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入

Module List Reviewers Committers
repo-cann/ops-math N/A loov1, wangrui_, songkai111, zhou-qilong, gubaocheng
*/*/op_graph/*_proto.h N/A wangyongguang, tang-lei01
*/*/op_host/*_def.cpp N/A wangyongguang, tang-lei01
likedislike
CANN-robotCANN-robot成员
1月27日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
1月27日 评论:

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

likedislike
huangyuqian
1月27日 评论:

compile

likedislike
此处折叠了136条消息 查看更多
CANN-robotCANN-robot成员
1月29日 添加了label:lgtm
tang-lei01成员
1月29日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
1月29日 添加了label:approved
CANN-robot
CANN-robot成员
1月29日 评论:

Review Guide

This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: songkai111, tang-lei01.
Reviewers who wrote a comment of /lgtm are: songkai111, wangrui_, tang-lei01.

likedislike
CANN-robotCANN-robot成员
1月29日 合入了pull request