From f79ed946e3b1b9a100f298117539445f3837afba Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 18 Sep 2020 22:11:29 +0800
Subject: [PATCH] fix(mgb/cambricon): support uint8 input for cambricon runtime
 opr

GitOrigin-RevId: 7c35662dddbddc7731678252b0877a7a9479b445
---
 src/cambricon/impl/cambricon_runtime_opr.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/cambricon/impl/cambricon_runtime_opr.cpp b/src/cambricon/impl/cambricon_runtime_opr.cpp
index 29d256eda..cedc7d2c0 100644
--- a/src/cambricon/impl/cambricon_runtime_opr.cpp
+++ b/src/cambricon/impl/cambricon_runtime_opr.cpp
@@ -72,8 +72,14 @@ cnrtDataType_t mgb_dtype_to_cnrt_dtype(DType data_type) {
             return CNRT_FLOAT32;
         case DTypeEnum::QuantizedS8:
             return CNRT_QUANT8;
+        case DTypeEnum::Quantized8Asymm:
+            return CNRT_QUANT8;
+        case DTypeEnum::Int8:
+            return CNRT_INT8;
         case DTypeEnum::Int32:
             return CNRT_INT32;
+        case DTypeEnum::Uint8:
+            return CNRT_UINT8;
         default:
             mgb_throw(MegBrainError,
                       "megbrain data type %s is not supported by cnrt.",
-- 
GitLab