Browse Source

fix(sdk): fix error caused by incorrect parsing u16/s16 inputs in lar

GitOrigin-RevId: 9486cf362d
release-1.7
Megvii Engine Team 3 years ago
parent
commit
dbb3232c31
1 changed files with 6 additions and 3 deletions
  1. +6
    -3
      sdk/load-and-run/src/mgblar.cpp

+ 6
- 3
sdk/load-and-run/src/mgblar.cpp View File

@@ -423,10 +423,13 @@ private:
}

const std::map<std::string, megdnn::DType> 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) {


Loading…
Cancel
Save