GRU算子CPU计算

1.芯片型号:X3派

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

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

你好,关于算子是否运行在BPU上看,是需要看组成GRU模块的各个算子是否满足BPU约束,建议看一下最新 OE2.6.2对应的算子支持与约束列表,不满足约束的话,看看是否可以进行一些等效算子的替换。

如果无法实现,可以提供一下期望运行在BPU上的加速算子需求,我这边记录一下。

GRU模块是个整体,我看了支持列表只支持CPU加速

期望GRU算子可以支持BPU加速

嗯嗯,在算子支持与约束列表中看到GRU算子啦,我记录一下您的需求,向上反馈一下。

对了,地平线工具链在持续迭代优化,为了给您提供更好的服务,希望您能抽出3分钟左右的时间,将您在使用工具链期间的感受和建议告诉我们,您的宝贵意见对我们很重要,非常感谢!问卷链接:https://wenjuan.feishu.cn/m/cfm?t=st64p6krU3Ji-yvhv

算子需求,也可以记录到这里哦,会进行整理统计并向上反馈的