[1/1] ml/cnxk: enable data caching for TVM models
Checks
Commit Message
Enabled data caching for TVM models with MRVL only layers.
Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
---
drivers/ml/cnxk/cn10k_ml_ops.c | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
Comments
On Sun, Jan 7, 2024 at 8:50 PM Srikanth Yalavarthi
<syalavarthi@marvell.com> wrote:
>
> Enabled data caching for TVM models with MRVL only layers.
>
> Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
> ---
Applied to dpdk-next-net-mrvl/for-main. Thanks
@@ -996,8 +996,13 @@ 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 && model->type == ML_CNXK_MODEL_TYPE_GLOW)
- ret = cn10k_ml_cache_model_data(cnxk_mldev, layer);
+ 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);
+ }
}
return ret;