Added support for 64-bit integer data type in model metadata.
Signed-off-by: Srikanth Yalavarthi <[email protected]>
---
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
