[PATCH v2 1/1] ml/cnxk: fix name of TVM model with single layer
Srikanth Yalavarthi
syalavarthi at marvell.com
Wed Nov 22 18:13:19 CET 2023
Name field of TVM model with single MRVL layer is
currently set to empty string. Update the name with
the field from metadata.
Fixes: 5cea2c67edfc ("ml/cnxk: update internal TVM model info structure")
Signed-off-by: Srikanth Yalavarthi <syalavarthi at marvell.com>
---
v2:
- Replaced rte_memcpy with strlcpy
drivers/ml/cnxk/mvtvm_ml_model.c | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/drivers/ml/cnxk/mvtvm_ml_model.c b/drivers/ml/cnxk/mvtvm_ml_model.c
index 3e06ea658bb..e3234ae4422 100644
--- a/drivers/ml/cnxk/mvtvm_ml_model.c
+++ b/drivers/ml/cnxk/mvtvm_ml_model.c
@@ -352,6 +352,10 @@ mvtvm_ml_model_info_set(struct cnxk_ml_dev *cnxk_mldev, struct cnxk_ml_model *mo
tvm_mrvl_model:
cn10k_ml_model_info_set(cnxk_mldev, model, &model->mvtvm.info,
&model->layer[0].glow.metadata);
+
+ metadata = &model->mvtvm.metadata;
+ strlcpy(info->name, metadata->model.name, TVMDP_NAME_STRLEN);
+
info->io_layout = RTE_ML_IO_LAYOUT_SPLIT;
}
--
2.42.0
More information about the dev
mailing list