建议使用以下浏览器,以获得最佳体验。 IE 9.0+以上版本 Chrome 31+ 谷歌浏览器 Firefox 30+ 火狐浏览器
请选择 进入手机版 | 继续访问电脑版
设置昵称

在此一键设置昵称,即可参与社区互动!

确定
我再想想
选择版块
直达楼层
标签
您还可以添加5个标签
  • 没有搜索到和“关键字”相关的标签
  • 云产品
  • 解决方案
  • 技术领域
  • 通用技术
  • 平台功能
取消

采纳成功

您已采纳当前回复为最佳回复

BubtLTK

发帖: 0粉丝: 0

发消息 + 关注

发表于2021年04月08日 15:18:07 134 6
直达本楼层的链接
楼主
显示全部楼层
[算子开发] 【众智】【CANN】ut测试时实现的算子不起作用

十分感谢您的帮助!

【功能模块】

argmin:计算输入Tensor指定维度的最小值的索引号

【操作步骤&问题现象】

1、通过原生TensorFlow的算子接口生成测试的期望数据。

2、ut测试代码可以成功读入期望数据,但执行算子后,输出的output还是初始化时的数据

执行算子前的output如下:

image.png

执行算子后的output如下:

image.png

算子实例化过程如下:

image.png

image.png

image.png

ut测试反馈如下(最后一行64:Failure定位到的是代码RUN_KERNEL):

image.png




举报
分享

分享文章到朋友圈

分享文章到微博
精彩回复

采纳成功

您已采纳当前回复为最佳回复

莫晓忧

发帖: 0粉丝: 1

级别 : 版主

发消息 + 关注

发表于2021年04月08日 20:35:48
直达本楼层的链接
显示全部楼层 最佳回复

我使用你的代码本地调试后发现你的ut下的cmakelists内的set(_cpu_kernels_src下没有添加ArgMin.cc,只需要添加一下源码文件路径就能编译通过了,如图:image.png

点赞 评论 引用 举报

采纳成功

您已采纳当前回复为最佳回复

莫晓忧

发帖: 0粉丝: 1

级别 : 版主

发消息 + 关注

发表于2021年04月08日 15:41:11
直达本楼层的链接
沙发
显示全部楼层

请问有在ops/built-in/aicpu/impl/CMakeLists.txt内配置编译算子实现文件么?

点赞 评论 引用 举报

采纳成功

您已采纳当前回复为最佳回复

BubtLTK

发帖: 0粉丝: 0

发消息 + 关注

发表于2021年04月08日 15:51:00
直达本楼层的链接
板凳
显示全部楼层

ops/built-in/aicpu/impl/CMakeLists.txt内已经配置了编译算子实现文件,下图中的arg_min.cc文件就是,已配置在代码段set(local_kernels_src_files......)中

image.png

点赞 评论 引用 举报

采纳成功

您已采纳当前回复为最佳回复

莫晓忧

发帖: 0粉丝: 1

级别 : 版主

发消息 + 关注

发表于2021年04月08日 15:55:13
直达本楼层的链接
地板
显示全部楼层

算子信息库内配置算子信息了么?

点赞 评论 引用 举报

采纳成功

您已采纳当前回复为最佳回复

BubtLTK

发帖: 0粉丝: 0

发消息 + 关注

发表于2021年04月08日 16:03:29
直达本楼层的链接
5#
显示全部楼层

算子信息库已配置算子信息,信息如下:

image.png

评论
莫晓忧 2021-4-8 16:49 评论

方便把ArgMin相关的修改文件发我一下么?我这边编译定位一下。

... 查看全部
莫晓忧 2021-4-8 17:06 评论

实现代码里有写REGISTER_CPU_KERNEL(argmin, ArgMinCpuKernel);去注册该算子实现么?

... 查看全部
BubtLTK 2021-4-8 18:20 评论

评论 莫晓忧:嗯,算子实现代码已注册。

... 查看全部
点赞 评论 引用 举报

采纳成功

您已采纳当前回复为最佳回复

zhujianpeng

发帖: 16粉丝: 1

级别 : 版主,版块专家

发消息 + 关注

发表于2021年04月08日 16:43:45
直达本楼层的链接
6#
显示全部楼层

mark


点赞 评论 引用 举报

游客

富文本
Markdown
您需要登录后才可以回帖 登录 | 立即注册

邀请回答
您可以邀请3位专家

结贴

您对问题的回复是否满意?
满意度
非常满意 满意 一般 不满意
我要反馈
0/200