File tree Expand file tree Collapse file tree 1 file changed +12
-10
lines changed Expand file tree Collapse file tree 1 file changed +12
-10
lines changed Original file line number Diff line number Diff line change @@ -80,15 +80,16 @@ inline VarDataType ConvertPrecisionType(lite_api::PrecisionType type) {
80
80
return lite::VarDataType::vtype; \
81
81
break
82
82
switch (type) {
83
- CASE (Float, FP32 );
83
+ CASE (Bool, BOOL );
84
84
CASE (Int8, INT8);
85
+ CASE (UInt8, UINT8);
86
+ CASE (Int16, INT16);
85
87
CASE (Int32, INT32);
86
- CASE (FP16, FP16);
87
- CASE (Bool, BOOL);
88
88
CASE (Int64, INT64);
89
- CASE (Int16, INT16);
89
+ CASE (FP16, FP16);
90
+ CASE (Float, FP32);
90
91
default :
91
- LOG (FATAL) << " Illegal flatbuffer VarType." ;
92
+ LOG (FATAL) << " Illegal flatbuffer VarType." << static_cast < int >(type) ;
92
93
return lite::VarDataType ();
93
94
}
94
95
#undef CASE
@@ -100,15 +101,16 @@ inline lite_api::PrecisionType ConvertPrecisionType(VarDataType type) {
100
101
return lite_api::PrecisionType::k##ptype; \
101
102
break
102
103
switch (type) {
103
- CASE (Float, FP32 );
104
+ CASE (Bool, BOOL );
104
105
CASE (Int8, INT8);
106
+ CASE (UInt8, UINT8);
107
+ CASE (Int16, INT16);
105
108
CASE (Int32, INT32);
106
- CASE (FP16, FP16);
107
- CASE (Bool, BOOL);
108
109
CASE (Int64, INT64);
109
- CASE (Int16, INT16);
110
+ CASE (FP16, FP16);
111
+ CASE (Float, FP32);
110
112
default :
111
- LOG (FATAL) << " Illegal flatbuffer VarType. " ;
113
+ LOG (FATAL) << " Illegal flatbuffer VarType: " << static_cast < int >(type) ;
112
114
return lite_api::PrecisionType ();
113
115
}
114
116
#undef CASE
You can’t perform that action at this time.
0 commit comments