diff --git a/sdk/load-and-run/src/mgblar.cpp b/sdk/load-and-run/src/mgblar.cpp index a7f71789..b3d901b7 100644 --- a/sdk/load-and-run/src/mgblar.cpp +++ b/sdk/load-and-run/src/mgblar.cpp @@ -423,10 +423,13 @@ private: } const std::map type_map = { - {"f4", dtype::Float32()}, + {"f4", dtype::Float32()}, {"i4", dtype::Int32()}, - {"i1", dtype::Int8()}, - {"u1", dtype::Uint8()}}; + {"u2", dtype::Uint16()}, + {"i2", dtype::Int16()}, + {"i1", dtype::Int8()}, + {"u1", dtype::Uint8()}, + }; megdnn::DType hv_type; for (auto& item : type_map) {