- //init
- rknn_devices_id devids;
- ret = rknn_find_devices(&devids);
- printf("n_devices=%d\n", devids.n_devices);
- int npu_id = thread_id % devids.n_devices;
- rknn_init_extend extend;
- extend.device_id = devids.ids[npu_id];
- rknn_context ctx = 0;
- ret = rknn_init2(&ctx,model,model_len,RKNN_FLAG_PRIOR_MEDIUM,&extend);
- // ret = rknn_init(&ctx,model,model_len,RKNN_FLAG_PRIOR_MEDIUM);
- if(ret < 0) {
- printf("rknn_init fail! ret=%d\n", ret);
- return;
- }
- else
- {
- printf("using npu:%d,%s \n", npu_id, devids.types[npu_id]);
- }
复制代码
欢迎光临 Toybrick (https://t.rock-chips.com/) | Powered by Discuz! X3.3 |