diff mbox series

[v5,31/34] ml/cnxk: add generic ML malloc and free callback

Message ID	20231018064806.24145-32-syalavarthi@marvell.com (mailing list archive)
State	Superseded, archived
Delegated to:	Jerin Jacob
Headers	From: Srikanth Yalavarthi <syalavarthi@marvell.com> To: Srikanth Yalavarthi <syalavarthi@marvell.com> CC: <dev@dpdk.org>, <sshankarnara@marvell.com>, <aprabhu@marvell.com>, <ptakkar@marvell.com> Subject: [PATCH v5 31/34] ml/cnxk: add generic ML malloc and free callback Date: Tue, 17 Oct 2023 23:47:59 -0700 Message-ID: <20231018064806.24145-32-syalavarthi@marvell.com> In-Reply-To: <20231018064806.24145-1-syalavarthi@marvell.com> References: <20230830155927.3566-1-syalavarthi@marvell.com> <20231018064806.24145-1-syalavarthi@marvell.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Content-Type: text/plain Precedence: list Errors-To: dev-bounces@dpdk.org
Series	Implementation of revised ml/cnxk driver \| [v5,00/34] Implementation of revised ml/cnxk driver [v5,01/34] ml/cnxk: drop support for register polling [v5,02/34] ml/cnxk: add generic cnxk device structure [v5,03/34] ml/cnxk: add generic model and layer structures [v5,04/34] ml/cnxk: add generic cnxk request structure [v5,05/34] ml/cnxk: add generic cnxk xstats structures [v5,06/34] ml/cnxk: rename cnxk ops function pointers struct [v5,07/34] ml/cnxk: update device handling functions [v5,08/34] ml/cnxk: update queue-pair handling functions [v5,09/34] ml/cnxk: update model load and unload functions [v5,10/34] ml/cnxk: update model start and stop functions [v5,11/34] ml/cnxk: update model utility functions [v5,12/34] ml/cnxk: update data quantization functions [v5,13/34] ml/cnxk: update device debug functions [v5,14/34] ml/cnxk: update device stats functions [v5,15/34] ml/cnxk: update device and model xstats functions [v5,16/34] ml/cnxk: update fast path functions [v5,17/34] ml/cnxk: move error handling to cnxk layer [v5,18/34] ml/cnxk: support config and close of tvmdp library [v5,19/34] ml/cnxk: add structures to support TVM model type [v5,20/34] ml/cnxk: add support for identify model type [v5,21/34] ml/cnxk: add support to parse TVM model objects [v5,22/34] ml/cnxk: fetch layer info and load TVM model [v5,23/34] ml/cnxk: update internal info for TVM model [v5,24/34] ml/cnxk: enable model unload in tvmdp library [v5,25/34] ml/cnxk: enable OCM check for multilayer TVM model [v5,26/34] ml/cnxk: support start and stop for TVM models [v5,27/34] ml/cnxk: update internal TVM model info structure [v5,28/34] ml/cnxk: support device dump for TVM models [v5,29/34] ml/cnxk: enable reporting model runtime as xstats [v5,30/34] ml/cnxk: implement I/O alloc and free callbacks [v5,31/34] ml/cnxk: add generic ML malloc and free callback [v5,32/34] ml/cnxk: support quantize and dequantize callback [v5,33/34] ml/cnxk: enable fast-path ops for TVM models [v5,34/34] ml/cnxk: enable creation of mvtvm virtual device

Checks

Context	Check	Description
ci/checkpatch	success	coding style OK

Commit Message

Srikanth Yalavarthi Oct. 18, 2023, 6:47 a.m. UTC

  Implemented generic ML malloc and free callbacks

Signed-off-by: Srikanth Yalavarthi <syalavarthi@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 mbox series

Patch

diff --git a/drivers/ml/cnxk/cn10k_ml_ops.c b/drivers/ml/cnxk/cn10k_ml_ops.c
index 7802425c87..01b0a44caa 100644
--- a/drivers/ml/cnxk/cn10k_ml_ops.c
+++ b/drivers/ml/cnxk/cn10k_ml_ops.c
@@ -1497,3 +1497,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 9c41c1c0b0..eb3e1c139c 100644
--- a/drivers/ml/cnxk/cn10k_ml_ops.h
+++ b/drivers/ml/cnxk/cn10k_ml_ops.h
@@ -333,6 +333,9 @@  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);
+
 /* xstats ops */
 void cn10k_ml_xstat_model_name_set(struct cnxk_ml_dev *cnxk_mldev, struct cnxk_ml_model *model,
 				   uint16_t stat_id, uint16_t entry, char *suffix);
diff --git a/drivers/ml/cnxk/mvtvm_ml_ops.c b/drivers/ml/cnxk/mvtvm_ml_ops.c
index 77c2b5bcdc..b627355917 100644
--- a/drivers/ml/cnxk/mvtvm_ml_ops.c
+++ b/drivers/ml/cnxk/mvtvm_ml_ops.c
@@ -234,6 +234,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;
 	}