[PATCH 1/1] ml/cnxk: support for 64-bit int type in metadata

[Srikanth Yalavarthi]

Added support for 64-bit integer data type in model metadata.

Signed-off-by: Srikanth Yalavarthi <syalavarthi at marvell.com>
---
 drivers/ml/cnxk/cn10k_ml_model.c | 4 ++++
 drivers/ml/cnxk/cn10k_ml_model.h | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/drivers/ml/cnxk/cn10k_ml_model.c b/drivers/ml/cnxk/cn10k_ml_model.c
index 12a2dda800e..3eb64825d24 100644
--- a/drivers/ml/cnxk/cn10k_ml_model.c
+++ b/drivers/ml/cnxk/cn10k_ml_model.c
@@ -33,6 +33,10 @@ cn10k_ml_io_type_map(uint8_t type)
 		return RTE_ML_IO_TYPE_FP16;
 	case 8:
 		return RTE_ML_IO_TYPE_FP32;
+	case 9:
+		return RTE_ML_IO_TYPE_INT64;
+	case 10:
+		return RTE_ML_IO_TYPE_UINT64;
 	}
 
 	return RTE_ML_IO_TYPE_UNKNOWN;
diff --git a/drivers/ml/cnxk/cn10k_ml_model.h b/drivers/ml/cnxk/cn10k_ml_model.h
index 6744175cd5e..746eed7c702 100644
--- a/drivers/ml/cnxk/cn10k_ml_model.h
+++ b/drivers/ml/cnxk/cn10k_ml_model.h
@@ -224,12 +224,14 @@ struct cn10k_ml_model_metadata_input_section {
 	/* Type of incoming input
 	 * 1 = INT8, 2 = UINT8, 3 = INT16, 4 = UINT16,
 	 * 5 = INT32, 6 = UINT32, 7 = FP16, 8 = FP32
+	 * 9 = INT64, 10 = UINT64
 	 */
 	uint8_t input_type;
 
 	/* Type of input required by model
 	 * 1 = INT8, 2 = UINT8, 3 = INT16, 4 = UINT16,
 	 * 5 = INT32, 6 = UINT32, 7 = FP16, 8 = FP32
+	 * 9 = INT64, 10 = UINT64
 	 */
 	uint8_t model_input_type;
 
@@ -282,12 +284,14 @@ struct cn10k_ml_model_metadata_output_section {
 	/* Type of outgoing output
 	 * 1 = INT8, 2 = UINT8, 3 = INT16, 4 = UINT16
 	 * 5 = INT32, 6 = UINT32, 7 = FP16, 8 = FP32
+	 * 9 = INT64, 10 = UINT64
 	 */
 	uint8_t output_type;
 
 	/* Type of output produced by model
 	 * 1 = INT8, 2 = UINT8, 3 = INT16, 4 = UINT16
 	 * 5 = INT32, 6 = UINT32, 7 = FP16, 8 = FP32
+	 * 9 = INT64, 10 = UINT64
 	 */
 	uint8_t model_output_type;
 
-- 
2.34.1