RK3588板上执行TopK算子会触发段错误

Jerry_Zh

  运行一个YOLO魔改模型时，发现其在虚拟机上可以进行RKNN转换并正确推理，但在鲁班猫RK3588开发板上使用rknntoolkit-lite2进行推理会触发段错误，通过对模型进行拆分，定位到是其中的TopK算子引发的段错误，rknntoolkit2版本是2.3.0，查看官方算子支持文档也支持TopK算子，但为什么不管是ADB连接使用rknntoolkit2连板推理还是使用rknntoolkit-lite2板上离线推理到TopK算子都会引发段错误？该TopK算子输入形状为(1,784,784)，输出values形状为(1,784,1),请问这种问题应该怎么解决和排查？

  附上测试用的代码，可放在RK3588开发板上直接运行。

monark

可以先看下，https://doc.embedfire.com/linux/ ... le/RT-DETR.html#id4，自定义算子。

Jerry_Zh

monark 发表于 2025-5-22 10:11
可以先看下，https://doc.embedfire.com/linux/rk356x/Ai/zh/latest/lubancat_ai/example/RT-DETR.html#id4 ...

好的，非常感谢！