From mboxrd@z Thu Jan 1 00:00:00 1970
Return-Path: <dev-bounces@dpdk.org>
Received: from mails.dpdk.org (mails.dpdk.org [217.70.189.124])
by inbox.dpdk.org (Postfix) with ESMTP id 14225A00C4;
Mon, 14 Nov 2022 13:08:20 +0100 (CET)
Received: from mails.dpdk.org (localhost [127.0.0.1])
by mails.dpdk.org (Postfix) with ESMTP id 8341F42C29;
Mon, 14 Nov 2022 13:08:19 +0100 (CET)
Received: from mx0b-0016f401.pphosted.com (mx0b-0016f401.pphosted.com
[67.231.156.173])
by mails.dpdk.org (Postfix) with ESMTP id 17F9B42C29
for <dev@dpdk.org>; Mon, 14 Nov 2022 13:08:17 +0100 (CET)
Received: from pps.filterd (m0045851.ppops.net [127.0.0.1])
by mx0b-0016f401.pphosted.com (8.17.1.19/8.17.1.19) with ESMTP id
2AE6hJAg008852; Mon, 14 Nov 2022 04:05:53 -0800
DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=marvell.com;
h=from : to : cc :
subject : date : message-id : in-reply-to : references : mime-version :
content-transfer-encoding : content-type; s=pfpt0220;
bh=NLlbwHmrW2i3Ymqu0ltU8fOPv5T2d+s5wwdLa/WaklU=;
b=BUGH0tZLXP8l0w6INszFFyInlVeZUvV9KFoFLf4XpKX51Kp71yKXhLmrotqVYZdhqJ0i
LtDyPRV2V+rA2jptj7/UxAY8u3r8Xg3IRSXL0+LaARImgBvA8TYd7hktvTjvOhU+yjDp
I8D6xlm/gI1aBfXqy36iz3z+S6mQ3g+UDM1u5Jg+vhRw9iFToSESxF09EHyABw754UsX
9SGNXu9a/hN5J/bQ7DDqRQxaMzqWHpzBAqTflBrPqnkqCEl+oWkoVIVUOomcnbsi0IBr
QUE+kKQz37ZAXpNUtifwh1p3MxvGOYEBJOyZ1pNVtXoOzX70KlJZHFy9BoDUW4msSpTL eA==
Received: from dc5-exch02.marvell.com ([199.233.59.182])
by mx0b-0016f401.pphosted.com (PPS) with ESMTPS id 3kugnb0wuh-1
(version=TLSv1.2 cipher=ECDHE-RSA-AES256-SHA384 bits=256 verify=NOT);
Mon, 14 Nov 2022 04:05:53 -0800
Received: from DC5-EXCH01.marvell.com (10.69.176.38) by DC5-EXCH02.marvell.com
(10.69.176.39) with Microsoft SMTP Server (TLS) id 15.0.1497.18;
Mon, 14 Nov 2022 04:05:51 -0800
Received: from maili.marvell.com (10.69.176.80) by DC5-EXCH01.marvell.com
(10.69.176.38) with Microsoft SMTP Server id 15.0.1497.2 via Frontend
Transport; Mon, 14 Nov 2022 04:05:51 -0800
Received: from jerin-lab.marvell.com (jerin-lab.marvell.com [10.28.34.14])
by maili.marvell.com (Postfix) with ESMTP id 5E0BE5C68E6;
Mon, 14 Nov 2022 04:05:26 -0800 (PST)
From: <jerinj@marvell.com>
To: <dev@dpdk.org>, Srikanth Yalavarthi <syalavarthi@marvell.com>
CC: <thomas@monjalon.net>, <ferruh.yigit@xilinx.com>,
<ajit.khaparde@broadcom.com>, <aboyer@pensando.io>,
<andrew.rybchenko@oktetlabs.ru>, <beilei.xing@intel.com>,
<bruce.richardson@intel.com>, <chas3@att.com>, <chenbo.xia@intel.com>,
<ciara.loftus@intel.com>, <dsinghrawat@marvell.com>,
<ed.czeck@atomicrules.com>, <evgenys@amazon.com>, <grive@u256.net>,
<g.singh@nxp.com>, <zhouguoyang@huawei.com>, <haiyue.wang@intel.com>,
<hkalra@marvell.com>, <heinrich.kuhn@corigine.com>,
<hemant.agrawal@nxp.com>, <hyonkim@cisco.com>, <igorch@amazon.com>,
<irusskikh@marvell.com>, <jgrajcia@cisco.com>,
<jasvinder.singh@intel.com>, <jianwang@trustnetic.com>,
<jiawenwu@trustnetic.com>, <jingjing.wu@intel.com>,
<johndale@cisco.com>, <john.miller@atomicrules.com>,
<linville@tuxdriver.com>, <keith.wiles@intel.com>,
<kirankumark@marvell.com>, <oulijun@huawei.com>, <lironh@marvell.com>,
<longli@microsoft.com>, <mw@semihalf.com>, <spinler@cesnet.cz>,
<matan@nvidia.com>, <matt.peters@windriver.com>,
<maxime.coquelin@redhat.com>, <mk@semihalf.com>, <humin29@huawei.com>,
<pnalla@marvell.com>, <ndabilpuram@marvell.com>,
<qiming.yang@intel.com>, <qi.z.zhang@intel.com>, <radhac@marvell.com>,
<rahul.lakkireddy@chelsio.com>, <rmody@marvell.com>,
<rosen.xu@intel.com>, <sachin.saxena@oss.nxp.com>,
<skoteshwar@marvell.com>, <shshaikh@marvell.com>,
<shaibran@amazon.com>, <shepard.siegel@atomicrules.com>,
<asomalap@amd.com>, <somnath.kotur@broadcom.com>,
<sthemmin@microsoft.com>, <steven.webster@windriver.com>,
<skori@marvell.com>, <mtetsuyah@gmail.com>, <vburru@marvell.com>,
<viacheslavo@nvidia.com>, <xiao.w.wang@intel.com>,
<cloud.wangxiaoyun@huawei.com>, <yisen.zhuang@huawei.com>,
<yongwang@vmware.com>, <xuanziyang2@huawei.com>, <pkapoor@marvell.com>,
<nadavh@marvell.com>, <sburla@marvell.com>, <pathreya@marvell.com>,
<gakhil@marvell.com>, <mdr@ashroe.eu>, <dmitry.kozliuk@gmail.com>,
<anatoly.burakov@intel.com>, <cristian.dumitrescu@intel.com>,
<honnappa.nagarahalli@arm.com>, <mattias.ronnblom@ericsson.com>,
<ruifeng.wang@arm.com>, <drc@linux.vnet.ibm.com>,
<konstantin.ananyev@intel.com>, <olivier.matz@6wind.com>,
<jay.jayatheerthan@intel.com>, <asekhar@marvell.com>,
<pbhagavatula@marvell.com>, <eagostini@nvidia.com>,
<dchickles@marvell.com>, <sshankarnara@marvell.com>,
Jerin Jacob <jerinj@marvell.com>
Subject: [dpdk-dev] [PATCH v1 06/12] mldev: support input and output data
handling
Date: Mon, 14 Nov 2022 17:32:32 +0530
Message-ID: <20221114120238.2143832-7-jerinj@marvell.com>
X-Mailer: git-send-email 2.38.1
In-Reply-To: <20221114120238.2143832-1-jerinj@marvell.com>
References: <20220803132839.2747858-2-jerinj@marvell.com>
<20221114120238.2143832-1-jerinj@marvell.com>
MIME-Version: 1.0
Content-Transfer-Encoding: 8bit
Content-Type: text/plain
X-Proofpoint-ORIG-GUID: so50KI0ZvCEpwdGb5yZTCSygYkDwFV3Z
X-Proofpoint-GUID: so50KI0ZvCEpwdGb5yZTCSygYkDwFV3Z
X-Proofpoint-Virus-Version: vendor=baseguard
engine=ICAP:2.0.219,Aquarius:18.0.895,Hydra:6.0.545,FMLib:17.11.122.1
definitions=2022-11-14_10,2022-11-11_01,2022-06-22_01
X-BeenThere: dev@dpdk.org
X-Mailman-Version: 2.1.29
Precedence: list
List-Id: DPDK patches and discussions <dev.dpdk.org>
List-Unsubscribe: <https://mails.dpdk.org/options/dev>,
<mailto:dev-request@dpdk.org?subject=unsubscribe>
List-Archive: <http://mails.dpdk.org/archives/dev/>
List-Post: <mailto:dev@dpdk.org>
List-Help: <mailto:dev-request@dpdk.org?subject=help>
List-Subscribe: <https://mails.dpdk.org/listinfo/dev>,
<mailto:dev-request@dpdk.org?subject=subscribe>
Errors-To: dev-bounces@dpdk.org
From: Srikanth Yalavarthi <syalavarthi@marvell.com>
Added RTE library functions to handle model input and
output data. The APIs can be used to get the size of I/O
buffers, quantize input data and dequantize output data.
Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
Signed-off-by: Jerin Jacob <jerinj@marvell.com>
---
lib/mldev/rte_mldev.c | 94 ++++++++++++++++++++++++++++++++
lib/mldev/rte_mldev_core.h | 106 +++++++++++++++++++++++++++++++++++++
lib/mldev/version.map | 4 ++
3 files changed, 204 insertions(+)
diff --git a/lib/mldev/rte_mldev.c b/lib/mldev/rte_mldev.c
index 327ed7144d..13b7e93943 100644
--- a/lib/mldev/rte_mldev.c
+++ b/lib/mldev/rte_mldev.c
@@ -462,3 +462,97 @@ rte_ml_model_params_update(int16_t dev_id, int16_t model_id, void *buffer)
return (*dev->dev_ops->model_params_update)(dev, model_id, buffer);
}
+
+int
+rte_ml_io_input_size_get(int16_t dev_id, int16_t model_id, uint32_t nb_batches,
+ uint64_t *input_qsize, uint64_t *input_dsize)
+{
+ struct rte_ml_dev *dev;
+
+ if (!rte_ml_dev_is_valid_dev(dev_id)) {
+ ML_DEV_LOG(ERR, "Invalid dev_id = %d\n", dev_id);
+ return -EINVAL;
+ }
+
+ dev = rte_ml_dev_pmd_get_dev(dev_id);
+ if (*dev->dev_ops->io_input_size_get == NULL)
+ return -ENOTSUP;
+
+ return (*dev->dev_ops->io_input_size_get)(dev, model_id, nb_batches, input_qsize,
+ input_dsize);
+}
+
+int
+rte_ml_io_output_size_get(int16_t dev_id, int16_t model_id, uint32_t nb_batches,
+ uint64_t *output_qsize, uint64_t *output_dsize)
+{
+ struct rte_ml_dev *dev;
+
+ if (!rte_ml_dev_is_valid_dev(dev_id)) {
+ ML_DEV_LOG(ERR, "Invalid dev_id = %d\n", dev_id);
+ return -EINVAL;
+ }
+
+ dev = rte_ml_dev_pmd_get_dev(dev_id);
+ if (*dev->dev_ops->io_output_size_get == NULL)
+ return -ENOTSUP;
+
+ return (*dev->dev_ops->io_output_size_get)(dev, model_id, nb_batches, output_qsize,
+ output_dsize);
+}
+
+int
+rte_ml_io_quantize(int16_t dev_id, int16_t model_id, uint16_t nb_batches, void *dbuffer,
+ void *qbuffer)
+{
+ struct rte_ml_dev *dev;
+
+ if (!rte_ml_dev_is_valid_dev(dev_id)) {
+ ML_DEV_LOG(ERR, "Invalid dev_id = %d\n", dev_id);
+ return -EINVAL;
+ }
+
+ dev = rte_ml_dev_pmd_get_dev(dev_id);
+ if (*dev->dev_ops->io_quantize == NULL)
+ return -ENOTSUP;
+
+ if (dbuffer == NULL) {
+ ML_DEV_LOG(ERR, "Dev %d, dbuffer cannot be NULL\n", dev_id);
+ return -EINVAL;
+ }
+
+ if (qbuffer == NULL) {
+ ML_DEV_LOG(ERR, "Dev %d, qbuffer cannot be NULL\n", dev_id);
+ return -EINVAL;
+ }
+
+ return (*dev->dev_ops->io_quantize)(dev, model_id, nb_batches, dbuffer, qbuffer);
+}
+
+int
+rte_ml_io_dequantize(int16_t dev_id, int16_t model_id, uint16_t nb_batches, void *qbuffer,
+ void *dbuffer)
+{
+ struct rte_ml_dev *dev;
+
+ if (!rte_ml_dev_is_valid_dev(dev_id)) {
+ ML_DEV_LOG(ERR, "Invalid dev_id = %d\n", dev_id);
+ return -EINVAL;
+ }
+
+ dev = rte_ml_dev_pmd_get_dev(dev_id);
+ if (*dev->dev_ops->io_dequantize == NULL)
+ return -ENOTSUP;
+
+ if (qbuffer == NULL) {
+ ML_DEV_LOG(ERR, "Dev %d, qbuffer cannot be NULL\n", dev_id);
+ return -EINVAL;
+ }
+
+ if (dbuffer == NULL) {
+ ML_DEV_LOG(ERR, "Dev %d, dbuffer cannot be NULL\n", dev_id);
+ return -EINVAL;
+ }
+
+ return (*dev->dev_ops->io_dequantize)(dev, model_id, nb_batches, qbuffer, dbuffer);
+}
diff --git a/lib/mldev/rte_mldev_core.h b/lib/mldev/rte_mldev_core.h
index 172454c2aa..b388553a96 100644
--- a/lib/mldev/rte_mldev_core.h
+++ b/lib/mldev/rte_mldev_core.h
@@ -259,6 +259,100 @@ typedef int (*mldev_model_info_get_t)(struct rte_ml_dev *dev, int16_t model_id,
*/
typedef int (*mldev_model_params_update_t)(struct rte_ml_dev *dev, int16_t model_id, void *buffer);
+/**
+ * @internal
+ *
+ * Get size of input buffers.
+ *
+ * @param dev
+ * ML device pointer.
+ * @param model_id
+ * Model ID to use.
+ * @param nb_batches
+ * Number of batches.
+ * @param input_qsize
+ * Size of quantized input.
+ * @param input_dsize
+ * Size of dequantized input.
+ *
+ * @return
+ * - 0 on success.
+ * - <0, error on failure.
+ */
+typedef int (*mldev_io_input_size_get_t)(struct rte_ml_dev *dev, int16_t model_id,
+ uint32_t nb_batches, uint64_t *input_qsize,
+ uint64_t *input_dsize);
+
+/**
+ * @internal
+ *
+ * Get size of output buffers.
+ *
+ * @param dev
+ * ML device pointer.
+ * @param model_id
+ * Model ID to use.
+ * @param nb_batches
+ * Number of batches.
+ * @param output_qsize
+ * Size of quantized output.
+ * @param output_dsize
+ * Size of dequantized output.
+ *
+ * @return
+ * - 0 on success.
+ * - <0, error on failure.
+ */
+typedef int (*mldev_io_output_size_get_t)(struct rte_ml_dev *dev, int16_t model_id,
+ uint32_t nb_batches, uint64_t *output_qsize,
+ uint64_t *output_dsize);
+
+/**
+ * @internal
+ *
+ * Quantize model data.
+ *
+ * @param dev
+ * ML device pointer.
+ * @param model_id
+ * Model ID to use.
+ * @param nb_batches
+ * Number of batches.
+ * @param dbuffer
+ * Pointer t de-quantized data buffer.
+ * @param qbuffer
+ * Pointer t de-quantized data buffer.
+ *
+ * @return
+ * - 0 on success.
+ * - <0, error on failure.
+ */
+typedef int (*mldev_io_quantize_t)(struct rte_ml_dev *dev, int16_t model_id, uint16_t nb_batches,
+ void *dbuffer, void *qbuffer);
+
+/**
+ * @internal
+ *
+ * Quantize model data.
+ *
+ * @param dev
+ * ML device pointer.
+ * @param model_id
+ * Model ID to use.
+ * @param nb_batches
+ * Number of batches.
+ * @param qbuffer
+ * Pointer t de-quantized data buffer.
+ * @param dbuffer
+ * Pointer t de-quantized data buffer.
+ *
+ * @return
+ * - 0 on success.
+ * - <0, error on failure.
+ */
+typedef int (*mldev_io_dequantize_t)(struct rte_ml_dev *dev, int16_t model_id, uint16_t nb_batches,
+ void *qbuffer, void *dbuffer);
+
/**
* @internal
*
@@ -303,6 +397,18 @@ struct rte_ml_dev_ops {
/** Update model params. */
mldev_model_params_update_t model_params_update;
+
+ /** Get input buffer size. */
+ mldev_io_input_size_get_t io_input_size_get;
+
+ /** Get output buffer size. */
+ mldev_io_output_size_get_t io_output_size_get;
+
+ /** Quantize data */
+ mldev_io_quantize_t io_quantize;
+
+ /** De-quantize data */
+ mldev_io_dequantize_t io_dequantize;
};
/**
diff --git a/lib/mldev/version.map b/lib/mldev/version.map
index 4459f02925..0b180020db 100644
--- a/lib/mldev/version.map
+++ b/lib/mldev/version.map
@@ -10,6 +10,10 @@ EXPERIMENTAL {
rte_ml_dev_socket_id;
rte_ml_dev_start;
rte_ml_dev_stop;
+ rte_ml_io_dequantize;
+ rte_ml_io_input_size_get;
+ rte_ml_io_output_size_get;
+ rte_ml_io_quantize;
rte_ml_model_info_get;
rte_ml_model_load;
rte_ml_model_params_update;
--
2.38.1