From: Anup Prabhu <apra...@marvell.com> Enabled data caching for all MRVL layers in TVM models
Signed-off-by: Anup Prabhu <apra...@marvell.com> --- drivers/ml/cnxk/cn10k_ml_ops.c | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/drivers/ml/cnxk/cn10k_ml_ops.c b/drivers/ml/cnxk/cn10k_ml_ops.c index 834e55e88e9..f6b2baa0df0 100644 --- a/drivers/ml/cnxk/cn10k_ml_ops.c +++ b/drivers/ml/cnxk/cn10k_ml_ops.c @@ -996,13 +996,8 @@ cn10k_ml_layer_start(void *device, uint16_t model_id, const char *layer_name) if (ret < 0) { cn10k_ml_layer_stop(device, model_id, layer_name); } else { - if (cn10k_mldev->cache_model_data) { - if ((model->type == ML_CNXK_MODEL_TYPE_GLOW && - model->subtype == ML_CNXK_MODEL_SUBTYPE_GLOW_MRVL) || - (model->type == ML_CNXK_MODEL_TYPE_TVM && - model->subtype == ML_CNXK_MODEL_SUBTYPE_TVM_MRVL)) - ret = cn10k_ml_cache_model_data(cnxk_mldev, layer); - } + if (cn10k_mldev->cache_model_data) + ret = cn10k_ml_cache_model_data(cnxk_mldev, layer); } return ret; -- 2.45.1