[4/4] ml/cnxk: add adapter dequeue function

Message ID 20240107154013.4676-5-syalavarthi@marvell.com (mailing list archive)
State Changes Requested, archived
Delegated to: Jerin Jacob
Headers
Series Implementation of CNXK ML event adapter driver |

Checks

Context Check Description
ci/checkpatch success coding style OK
ci/loongarch-compilation fail ninja build failure

Commit Message

Srikanth Yalavarthi Jan. 7, 2024, 3:40 p.m. UTC
  Implemented ML adapter dequeue function.

Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
---
 drivers/event/cnxk/cn10k_worker.h   |  3 +++
 drivers/ml/cnxk/cn10k_ml_event_dp.h |  2 ++
 drivers/ml/cnxk/cn10k_ml_ops.c      | 17 +++++++++++++++++
 drivers/ml/cnxk/version.map         |  1 +
 4 files changed, 23 insertions(+)
  

Comments

Jerin Jacob Feb. 2, 2024, 9 a.m. UTC | #1
On Sun, Jan 7, 2024 at 11:39 PM Srikanth Yalavarthi
<syalavarthi@marvell.com> wrote:
>
> Implemented ML adapter dequeue function.
>
> Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>

Update the release notes  for this new feature in PMD section.
  

Patch

diff --git a/drivers/event/cnxk/cn10k_worker.h b/drivers/event/cnxk/cn10k_worker.h
index 8aa916fa129..1a0ca7f9493 100644
--- a/drivers/event/cnxk/cn10k_worker.h
+++ b/drivers/event/cnxk/cn10k_worker.h
@@ -7,6 +7,7 @@ 
 
 #include <rte_eventdev.h>
 #include "cn10k_cryptodev_event_dp.h"
+#include "cn10k_ml_event_dp.h"
 #include "cn10k_rx.h"
 #include "cnxk_worker.h"
 #include "cn10k_eventdev.h"
@@ -236,6 +237,8 @@  cn10k_sso_hws_post_process(struct cn10k_sso_hws *ws, uint64_t *u64,
 		/* Mark vector mempool object as get */
 		RTE_MEMPOOL_CHECK_COOKIES(rte_mempool_from_obj((void *)u64[1]),
 					  (void **)&u64[1], 1, 1);
+	} else if (CNXK_EVENT_TYPE_FROM_TAG(u64[0]) == RTE_EVENT_TYPE_MLDEV) {
+		u64[1] = cn10k_ml_adapter_dequeue(u64[1]);
 	}
 }
 
diff --git a/drivers/ml/cnxk/cn10k_ml_event_dp.h b/drivers/ml/cnxk/cn10k_ml_event_dp.h
index bf7fc57bceb..0ff92091296 100644
--- a/drivers/ml/cnxk/cn10k_ml_event_dp.h
+++ b/drivers/ml/cnxk/cn10k_ml_event_dp.h
@@ -12,5 +12,7 @@ 
 
 __rte_internal
 __rte_hot uint16_t cn10k_ml_adapter_enqueue(void *ws, struct rte_event ev[], uint16_t nb_events);
+__rte_internal
+__rte_hot uintptr_t cn10k_ml_adapter_dequeue(uintptr_t get_work1);
 
 #endif /* _CN10K_ML_EVENT_DP_H_ */
diff --git a/drivers/ml/cnxk/cn10k_ml_ops.c b/drivers/ml/cnxk/cn10k_ml_ops.c
index 4bc17eaa8c4..c33a7a85987 100644
--- a/drivers/ml/cnxk/cn10k_ml_ops.c
+++ b/drivers/ml/cnxk/cn10k_ml_ops.c
@@ -1660,3 +1660,20 @@  cn10k_ml_adapter_enqueue(void *ws, struct rte_event ev[], uint16_t nb_events)
 
 	return count;
 }
+
+__rte_hot uintptr_t
+cn10k_ml_adapter_dequeue(uintptr_t get_work1)
+{
+	struct cnxk_ml_dev *cnxk_mldev;
+	struct cnxk_ml_req *req;
+	struct cnxk_ml_qp *qp;
+
+	req = (struct cnxk_ml_req *)(get_work1);
+	cnxk_mldev = req->cnxk_mldev;
+	qp = cnxk_mldev->mldev->data->queue_pairs[req->qp_id];
+
+	cn10k_ml_result_update(cnxk_mldev, req->qp_id, req);
+	rte_mempool_put(qp->mla.req_mp, req);
+
+	return (uintptr_t)req->op;
+}
diff --git a/drivers/ml/cnxk/version.map b/drivers/ml/cnxk/version.map
index c2cacaf8c65..97c2c149998 100644
--- a/drivers/ml/cnxk/version.map
+++ b/drivers/ml/cnxk/version.map
@@ -2,6 +2,7 @@  INTERNAL {
 	global:
 
 	cn10k_ml_adapter_enqueue;
+	cn10k_ml_adapter_dequeue;
 
 	local: *;
 };