专栏算法工具链GRU算子CPU计算

GRU算子CPU计算

已解决

Everglow2023-07-22

34

0

0

5

1.芯片型号：X3派

3.问题定位：例如模型转换、板端部署等

我的模型里有个GRU模块，在X3派的算子支持列表上是CPU计算，请问有什么BPU加速的模块可以替换GRU模块或者可以把GRU模块转换为BPU加速吗？

算法工具链

评论1

0/1000

颜值即正义
Lv.2
你好，关于算子是否运行在BPU上看，是需要看组成GRU模块的各个算子是否满足BPU约束，建议看一下最新 OE2.6.2对应的算子支持与约束列表，不满足约束的话，看看是否可以进行一些等效算子的替换。
如果无法实现，可以提供一下期望运行在BPU上的加速算子需求，我这边记录一下。
2023-07-24
0
4
- Everglow回复颜值即正义:
  GRU模块是个整体，我看了支持列表只支持CPU加速
  2023-07-24
  0
  回复
- Everglow回复颜值即正义:
  期望GRU算子可以支持BPU加速
  2023-07-24
  0
  回复
- 颜值即正义回复Everglow:
  嗯嗯，在算子支持与约束列表中看到GRU算子啦，我记录一下您的需求，向上反馈一下。
  对了，地平线工具链在持续迭代优化，为了给您提供更好的服务，希望您能抽出3分钟左右的时间，将您在使用工具链期间的感受和建议告诉我们，您的宝贵意见对我们很重要，非常感谢！问卷链接：https://wenjuan.feishu.cn/m/cfm?t=st64p6krU3Ji-yvhv
  2023-07-24
  0
  回复
- 颜值即正义回复Everglow:
  算子需求，也可以记录到这里哦，会进行整理统计并向上反馈的
  2023-07-24
  0
  回复

暂无职位信息

0博客

2帖子

2回答