From fea5cf62797bbfc25e03b181c508b460618812fd Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 14 Oct 2020 22:26:30 +0800 Subject: [PATCH] fix(sdk/loader): debug sstar loader GitOrigin-RevId: a5847ad460e4bed5508a6ad113586264ecd82526 --- .../include/megbrain/serialization/extern_c_opr.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/serialization/include/megbrain/serialization/extern_c_opr.h b/src/serialization/include/megbrain/serialization/extern_c_opr.h index 9795cc87..4e3fdb8a 100644 --- a/src/serialization/include/megbrain/serialization/extern_c_opr.h +++ b/src/serialization/include/megbrain/serialization/extern_c_opr.h @@ -154,6 +154,9 @@ static inline size_t mgb_get_dtype_size(MGBDType dtype) { return 4; case MGB_DTYPE_UINT8: return 1; + case MGB_DTYPE_FLOAT16: + case MGB_DTYPE_INT16: + return 2; default: __builtin_trap(); return -1; -- GitLab