[PATCH v2 31/34] ml/cnxk: add generic ML malloc and free callback

[Srikanth Yalavarthi]

Implemented generic ML malloc and free callbacks

Signed-off-by: Srikanth Yalavarthi <syalavar...@marvell.com>
---
 drivers/ml/cnxk/cn10k_ml_ops.c | 30 ++++++++++++++++++++++++++++++
 drivers/ml/cnxk/cn10k_ml_ops.h |  3 +++
 drivers/ml/cnxk/mvtvm_ml_ops.c |  2 ++
 3 files changed, 35 insertions(+)

diff --git a/drivers/ml/cnxk/cn10k_ml_ops.c b/drivers/ml/cnxk/cn10k_ml_ops.c
index 23e98b96c5..140f7a343f 100644
--- a/drivers/ml/cnxk/cn10k_ml_ops.c
+++ b/drivers/ml/cnxk/cn10k_ml_ops.c
@@ -1522,3 +1522,33 @@ cn10k_ml_io_free(void *device, uint16_t model_id, const 
char *layer_name)
 
        return plt_memzone_free(mz);
 }
+
+int
+cn10k_ml_malloc(const char *name, size_t size, uint32_t align, void **addr)
+{
+       const struct plt_memzone *mz;
+
+       mz = plt_memzone_reserve_aligned(name, size, 0, align);
+       if (mz == NULL) {
+               plt_err("ml_malloc failed: Unable to allocate memory: name = 
%s", name);
+               return -ENOMEM;
+       }
+
+       *addr = mz->addr;
+
+       return 0;
+}
+
+int
+cn10k_ml_free(const char *name)
+{
+       const struct plt_memzone *mz;
+
+       mz = plt_memzone_lookup(name);
+       if (mz == NULL) {
+               plt_err("ml_free failed: Memzone not found: name = %s", name);
+               return -EINVAL;
+       }
+
+       return plt_memzone_free(mz);
+}
diff --git a/drivers/ml/cnxk/cn10k_ml_ops.h b/drivers/ml/cnxk/cn10k_ml_ops.h
index 055651eaa2..d7df1d003a 100644
--- a/drivers/ml/cnxk/cn10k_ml_ops.h
+++ b/drivers/ml/cnxk/cn10k_ml_ops.h
@@ -332,4 +332,7 @@ int cn10k_ml_io_alloc(void *device, uint16_t model_id, 
const char *layer_name,
                      uint64_t **input_qbuffer, uint64_t **output_qbuffer);
 int cn10k_ml_io_free(void *device, uint16_t model_id, const char *layer_name);
 
+int cn10k_ml_malloc(const char *name, size_t size, uint32_t align, void 
**addr);
+int cn10k_ml_free(const char *name);
+
 #endif /* _CN10K_ML_OPS_H_ */
diff --git a/drivers/ml/cnxk/mvtvm_ml_ops.c b/drivers/ml/cnxk/mvtvm_ml_ops.c
index a41ba4d343..95238d43d8 100644
--- a/drivers/ml/cnxk/mvtvm_ml_ops.c
+++ b/drivers/ml/cnxk/mvtvm_ml_ops.c
@@ -166,6 +166,8 @@ mvtvm_ml_model_load(struct cnxk_ml_dev *cnxk_mldev, struct 
rte_ml_model_params *
                callback->tvmrt_glow_layer_unload = cn10k_ml_layer_unload;
                callback->tvmrt_io_alloc = cn10k_ml_io_alloc;
                callback->tvmrt_io_free = cn10k_ml_io_free;
+               callback->tvmrt_malloc = cn10k_ml_malloc;
+               callback->tvmrt_free = cn10k_ml_free;
        } else {
                callback = NULL;
        }
-- 
2.41.0