Skip to content

Commit 59c658e

Browse files
committed
Merge branch 'master' of sipeed_ai/MaixPy
2 parents 2382795 + 31603f5 commit 59c658e

File tree

1 file changed

+21
-10
lines changed

1 file changed

+21
-10
lines changed

ports/k210-freertos/mpy_support/Maix/Maix_kpu.c

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,9 @@ int model_deinit(kpu_task_t *task)
294294

295295
STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_t *kw_args)
296296
{
297-
int err = 0;
297+
// int err = 0;
298+
char *err = NULL;
299+
298300
py_kpu_net_obj_t *o = m_new_obj(py_kpu_net_obj_t);
299301

300302
uint8_t *model_data = NULL;
@@ -304,7 +306,8 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
304306
kpu_task = (uint8_t *)malloc(sizeof(kpu_model_context_t));
305307
if(kpu_task == NULL)
306308
{
307-
err = -1;//malloc error
309+
// err = -1;//malloc error
310+
err = "malloc error 0";
308311
goto error;
309312
}
310313

@@ -326,28 +329,32 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
326329

327330
if(model_size < 0)
328331
{
329-
err = -2;//read error
332+
// err = -2;//read error
333+
err = "read model size error 1";
330334
goto error;
331335
}
332336
mp_printf(&mp_plat_print, "model_size=%d\r\n",model_size);
333337

334338
model_data = (uint8_t *)malloc(model_size * sizeof(uint8_t));
335339
if(model_data == NULL)
336340
{
337-
err = -1;//malloc error
341+
// err = -1;//malloc error
342+
err = "malloc error 1";
338343
goto error;
339344
}
340345

341346
status = w25qxx_read_data_dma(model_addr, model_data, model_size, W25QXX_QUAD_FAST);
342347
if(status != W25QXX_OK)
343348
{
344-
err = -2;//read error
349+
// err = -2;//read error
350+
err = "read model data error 1";
345351
goto error;
346352
}
347353
int ret = kpu_load_kmodel(kpu_task, model_data);
348354
if(ret != 0)
349355
{
350-
err = -3; //load error
356+
// err = -3; //load error
357+
err = "load model data error 1";
351358
goto error;
352359
}
353360
}
@@ -364,6 +371,7 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
364371
if( err != 0 )
365372
{
366373
model_deinit(kpu_task);
374+
err = "model init error 2";
367375
goto error;
368376
}
369377

@@ -382,15 +390,17 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
382390

383391
if(model_size <= 0)
384392
{
385-
err = -2;//read error
393+
// err = -2;//read error
394+
err = "read model size error 2";
386395
vfs_internal_close(file, &ferr);
387396
goto error;
388397
}
389398

390399
model_data = (uint8_t *)malloc(model_size * sizeof(uint8_t));
391400
if(model_data == NULL)
392401
{
393-
err = -1;//malloc error
402+
// err = -1;//malloc error
403+
err = "malloc error 2";
394404
goto error;
395405
}
396406

@@ -409,7 +419,8 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
409419
int ret = kpu_load_kmodel(kpu_task, model_data);
410420
if(ret != 0)
411421
{
412-
err = -3; //load error
422+
// err = -3; //load error
423+
err = "load model data error 2";
413424
goto error;
414425
}
415426

@@ -450,7 +461,7 @@ STATIC mp_obj_t py_kpu_class_load(uint n_args, const mp_obj_t *pos_args, mp_map_
450461
m_del(py_kpu_net_obj_t, o,sizeof(py_kpu_net_obj_t));
451462

452463
char msg[50];
453-
sprintf(msg,"[MAIXPY]kpu: load error %d", err);
464+
sprintf(msg,"[MAIXPY]:kpu load error: %s", err);
454465
mp_raise_ValueError(msg);
455466
return mp_const_false;
456467
}

0 commit comments

Comments
 (0)