[v1,1/1] ml/cnxk: fix updating internal I/O info
Checks
Commit Message
Update scale factor in IO info of TVM models from metadata.
Fixes: 35c3e790b4a0 ("ml/cnxk: update internal info for TVM model")
Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
---
drivers/ml/cnxk/mvtvm_ml_model.c | 2 ++
1 file changed, 2 insertions(+)
Comments
> -----Original Message-----
> From: Srikanth Yalavarthi <syalavarthi@marvell.com>
> Sent: Sunday, October 29, 2023 6:55 PM
> To: Srikanth Yalavarthi <syalavarthi@marvell.com>
> Cc: dev@dpdk.org; Shivah Shankar Shankar Narayan Rao
> <sshankarnara@marvell.com>; Anup Prabhu <aprabhu@marvell.com>;
> Prince Takkar <ptakkar@marvell.com>; Jerin Jacob Kollanukkaran
> <jerinj@marvell.com>
> Subject: [PATCH v1 1/1] ml/cnxk: fix updating internal I/O info
>
> Update scale factor in IO info of TVM models from metadata.
>
> Fixes: 35c3e790b4a0 ("ml/cnxk: update internal info for TVM model")
>
> Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
Acked-by: Shivah Shankar S <sshankarnara@marvell.com>
On Mon, Nov 6, 2023 at 8:43 PM Shivah Shankar Shankar Narayan Rao
<sshankarnara@marvell.com> wrote:
>
> > -----Original Message-----
> > From: Srikanth Yalavarthi <syalavarthi@marvell.com>
> > Sent: Sunday, October 29, 2023 6:55 PM
> > To: Srikanth Yalavarthi <syalavarthi@marvell.com>
> > Cc: dev@dpdk.org; Shivah Shankar Shankar Narayan Rao
> > <sshankarnara@marvell.com>; Anup Prabhu <aprabhu@marvell.com>;
> > Prince Takkar <ptakkar@marvell.com>; Jerin Jacob Kollanukkaran
> > <jerinj@marvell.com>
> > Subject: [PATCH v1 1/1] ml/cnxk: fix updating internal I/O info
> >
> > Update scale factor in IO info of TVM models from metadata.
> >
> > Fixes: 35c3e790b4a0 ("ml/cnxk: update internal info for TVM model")
> >
> > Signed-off-by: Srikanth Yalavarthi <syalavarthi@marvell.com>
> Acked-by: Shivah Shankar S <sshankarnara@marvell.com>
Applied to dpdk-next-net-mrvl/for-next-net. Thanks
@@ -216,6 +216,7 @@ mvtvm_ml_model_io_info_set(struct cnxk_ml_model *model)
model->mvtvm.info.input[i].sz_q =
model->mvtvm.info.input[i].nb_elements *
rte_ml_io_type_size_get(model->mvtvm.info.input[i].qtype);
+ model->mvtvm.info.input[i].scale = metadata->input[i].scale;
model->mvtvm.info.total_input_sz_d += model->mvtvm.info.input[i].sz_d;
model->mvtvm.info.total_input_sz_q += model->mvtvm.info.input[i].sz_q;
@@ -256,6 +257,7 @@ mvtvm_ml_model_io_info_set(struct cnxk_ml_model *model)
model->mvtvm.info.output[i].sz_q =
model->mvtvm.info.output[i].nb_elements *
rte_ml_io_type_size_get(model->mvtvm.info.output[i].qtype);
+ model->mvtvm.info.output[i].scale = metadata->output[i].scale;
model->mvtvm.info.total_output_sz_d += model->mvtvm.info.output[i].sz_d;
model->mvtvm.info.total_output_sz_q += model->mvtvm.info.output[i].sz_q;