Toybrick

标题: 关于文档里面支持的跑多核npu时出现的问题 [打印本页]

作者: roy 时间: 2023-4-7 14:38
标题: 关于文档里面支持的跑多核npu时出现的问题
运行代码：
rknn_core_mask core_mask = RKNN_NPU_CORE_0_1_2;
model = load_model(model_path, &model_len);
ret = rknn_init(&ctx, model, model_len, RKNN_FLAG_MEM_ALLOC_OUTSIDE, NULL);
int eeee = rknn_set_core_mask(ctx, core_mask);

报错：
E RKNN: [06:34:40.715] Not support core mask: 7, fallback to single core auto mode
E RKNN: [06:34:40.715] NN Compiler/Model Version is 1.3.0 now
E RKNN: [06:34:40.715] rknn_set_core_mask: failed to set core mask: 7

环境：
RK3588 Debian11 用的最新的固件5.10.110和自己编译的5.10.66都试过
rknn的so库是从github上拉的，github上的readme已经写到了release 1.4，但报错里面显示1.3，是不是库还是有问题？
想用3核对一个模型进行卷积加速优化的，卷积网络的权重用的一层高斯核测试的。

作者: roy 时间: 2023-4-7 14:42
本帖最后由 roy 于 2023-4-7 15:57 编辑

补充：内核一：rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.6.4内核二：rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.8.2

均报下面错误
E RKNN: [07:56:04.676] Not support core mask: 7, fallback to single core auto mode
E RKNN: [07:56:04.676] NN Compiler/Model Version is 1.2.6 now
E RKNN: [07:56:04.676] rknn_set_core_mask: failed to set core mask: 7

作者: wudiking 时间: 2023-4-10 06:39

roy 发表于 2023-4-7 14:42
补充：内核一：rknn_api/rknnrt version: 1.4.0 (a10f100eb@2022-09-09T09:07:14), driver version: 0.6.4 ...

把rknn的模型用1.4.0后的版本重新转换一下，我之前用过老模型就是这个问题，重新转一下就好了

作者: roy 时间: 2023-4-11 11:20

wudiking 发表于 2023-4-10 06:39
把rknn的模型用1.4.0后的版本重新转换一下，我之前用过老模型就是这个问题，重新转一下就好了 ...

感谢！已经解决，确实是这样

欢迎光临 Toybrick (https://t.rock-chips.com/)