[PATCH v1 1/1] mldev: support get queue pair count

[PATCH v1 1/1] mldev: support get queue pair count

[Srikanth Yalavarthi]

Added API to get ML device queue-pair count.

Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
---
 lib/mldev/rte_mldev.c | 15 +++++++++++++++
 lib/mldev/rte_mldev.h | 13 +++++++++++++
 lib/mldev/version.map |  1 +
 3 files changed, 29 insertions(+)

diff --git a/lib/mldev/rte_mldev.c b/lib/mldev/rte_mldev.c
index 196b1850e65..37d71aebc92 100644
--- a/lib/mldev/rte_mldev.c
+++ b/lib/mldev/rte_mldev.c
@@ -365,6 +365,21 @@ rte_ml_dev_stop(int16_t dev_id)
 	return ret;
 }
 
+uint16_t
+rte_ml_dev_queue_pair_count(int16_t dev_id)
+{
+	struct rte_ml_dev *dev;
+
+	if (!rte_ml_dev_is_valid_dev(dev_id)) {
+		RTE_MLDEV_LOG(ERR, "Invalid dev_id = %d", dev_id);
+		return -EINVAL;
+	}
+
+	dev = rte_ml_dev_pmd_get_dev(dev_id);
+
+	return dev->data->nb_queue_pairs;
+}
+
 int
 rte_ml_dev_queue_pair_setup(int16_t dev_id, uint16_t queue_pair_id,
 			    const struct rte_ml_dev_qp_conf *qp_conf, int socket_id)
diff --git a/lib/mldev/rte_mldev.h b/lib/mldev/rte_mldev.h
index 634af3d5e1a..767f8c2593b 100644
--- a/lib/mldev/rte_mldev.h
+++ b/lib/mldev/rte_mldev.h
@@ -317,6 +317,19 @@ struct rte_ml_dev_qp_conf {
 	 */
 };
 
+/**
+ * Get the number of queue pairs on a specific ML device.
+ *
+ * @param dev_id
+ *   The identifier of the device.
+ *
+ * @return
+ *   - The number of configured queue pairs.
+ */
+__rte_experimental
+uint16_t
+rte_ml_dev_queue_pair_count(int16_t dev_id);
+
 /**
  * Set up a queue pair for a device. This should only be called when the device is stopped.
  *
diff --git a/lib/mldev/version.map b/lib/mldev/version.map
index 84bdd6c3004..acf6395b4aa 100644
--- a/lib/mldev/version.map
+++ b/lib/mldev/version.map
@@ -11,6 +11,7 @@ EXPERIMENTAL {
 	rte_ml_dev_init;
 	rte_ml_dev_is_valid_dev;
 	rte_ml_dev_logtype;
+	rte_ml_dev_queue_pair_count;
 	rte_ml_dev_queue_pair_setup;
 	rte_ml_dev_selftest;
 	rte_ml_dev_socket_id;
-- 
2.45.1

Re: [PATCH v1 1/1] mldev: support get queue pair count

[Thomas Monjalon]

31/07/2024 08:13, Srikanth Yalavarthi:
> Added API to get ML device queue-pair count.
> 
> Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>

Applied, thanks.