Branch data Line data Source code
1 : : /* SPDX-License-Identifier: BSD-3-Clause
2 : : * Copyright (c) 2022 Marvell.
3 : : */
4 : :
5 : : #include <errno.h>
6 : : #include <stdint.h>
7 : :
8 : : #include <rte_mldev.h>
9 : : #include <rte_string_fns.h>
10 : :
11 : : #include "mldev_utils.h"
12 : :
13 : : /* Description:
14 : : * This file implements Machine Learning utility routines, except type conversion routines.
15 : : */
16 : :
17 : : int
18 [ # # ]: 0 : rte_ml_io_type_size_get(enum rte_ml_io_type type)
19 : : {
20 : : switch (type) {
21 : : case RTE_ML_IO_TYPE_UNKNOWN:
22 : : return -EINVAL;
23 : : case RTE_ML_IO_TYPE_INT8:
24 : : return sizeof(int8_t);
25 : : case RTE_ML_IO_TYPE_UINT8:
26 : : return sizeof(uint8_t);
27 : : case RTE_ML_IO_TYPE_INT16:
28 : : return sizeof(int16_t);
29 : : case RTE_ML_IO_TYPE_UINT16:
30 : : return sizeof(uint16_t);
31 : : case RTE_ML_IO_TYPE_INT32:
32 : : return sizeof(int32_t);
33 : : case RTE_ML_IO_TYPE_UINT32:
34 : : return sizeof(uint32_t);
35 : : case RTE_ML_IO_TYPE_FP8:
36 : : return sizeof(uint8_t);
37 : : case RTE_ML_IO_TYPE_FP16:
38 : : return sizeof(uint8_t) * 2;
39 : : case RTE_ML_IO_TYPE_FP32:
40 : : return sizeof(uint8_t) * 4;
41 : : case RTE_ML_IO_TYPE_BFLOAT16:
42 : : return sizeof(uint8_t) * 2;
43 : : default:
44 : : return -EINVAL;
45 : : }
46 : : }
47 : :
48 : : void
49 : 0 : rte_ml_io_type_to_str(enum rte_ml_io_type type, char *str, int len)
50 : : {
51 [ # # # # : 0 : switch (type) {
# # # # #
# # # ]
52 : 0 : case RTE_ML_IO_TYPE_UNKNOWN:
53 : 0 : rte_strlcpy(str, "unknown", len);
54 : : break;
55 : 0 : case RTE_ML_IO_TYPE_INT8:
56 : 0 : rte_strlcpy(str, "int8", len);
57 : : break;
58 : 0 : case RTE_ML_IO_TYPE_UINT8:
59 : 0 : rte_strlcpy(str, "uint8", len);
60 : : break;
61 : 0 : case RTE_ML_IO_TYPE_INT16:
62 : 0 : rte_strlcpy(str, "int16", len);
63 : : break;
64 : 0 : case RTE_ML_IO_TYPE_UINT16:
65 : 0 : rte_strlcpy(str, "uint16", len);
66 : : break;
67 : 0 : case RTE_ML_IO_TYPE_INT32:
68 : 0 : rte_strlcpy(str, "int32", len);
69 : : break;
70 : 0 : case RTE_ML_IO_TYPE_UINT32:
71 : 0 : rte_strlcpy(str, "uint32", len);
72 : : break;
73 : 0 : case RTE_ML_IO_TYPE_FP8:
74 : 0 : rte_strlcpy(str, "float8", len);
75 : : break;
76 : 0 : case RTE_ML_IO_TYPE_FP16:
77 : 0 : rte_strlcpy(str, "float16", len);
78 : : break;
79 : 0 : case RTE_ML_IO_TYPE_FP32:
80 : 0 : rte_strlcpy(str, "float32", len);
81 : : break;
82 : 0 : case RTE_ML_IO_TYPE_BFLOAT16:
83 : 0 : rte_strlcpy(str, "bfloat16", len);
84 : : break;
85 : 0 : default:
86 : 0 : rte_strlcpy(str, "invalid", len);
87 : : }
88 : 0 : }
|