当我通过udo算子实现平台不支持的算子时,如果我在json中不给我的算子设置scalar_params参数时,生成DLC是正常的。但只要加上......配置,生成dlc时就会报错, 应该如何处理:

shjky 2024-07-12 10:45:10

当我通过udo算子实现平台不支持的算子时,如果我在json中不给我的算子设置scalar_params参数时,生成DLC是正常的。但只要加上{"name": "mode", "data_type": "STRING"}, {"name": "padding_mode", "data_type": "STRING"} 配置,生成dlc时就会报错, 应该如何处理:

 

 

...全文
1207 1 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
weixin_38498942 2024-07-12
  • 打赏
  • 举报
回复

我们来举一个例子, 如果我们对SNPE 的UDO的定义如下图:

img

我们可以看到, 如果添加了字符串之后, 就会出现类似的错误, 这应该是scalar parameter不支持string类型,因为scalar本意就是没有dimension的数字。我们能够查到的所有udo例子中scalar parameter的type都是int或者float。我建议您尝试在json中把string类型的parameter映射成int来表示(比如0代表"mode0",1代表"mode1"),然后在实现udo的代码里用类似enum的数据类型或是数据结构来把读取到的int重新映射回您想要的string。

3,299

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧