diff --git a/src/cambricon/impl/cambricon_runtime_opr.cpp b/src/cambricon/impl/cambricon_runtime_opr.cpp index 29d256eda04fa6c3cf427f7c175efb4b0a01af0b..cedc7d2c0086e3ea9815135120870d70493acc66 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.",