提交 f79ed946 编写于 作者: M Megvii Engine Team

fix(mgb/cambricon): support uint8 input for cambricon runtime opr

GitOrigin-RevId: 7c35662dddbddc7731678252b0877a7a9479b445
上级 0b4918b2
......@@ -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.",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册