Browse Source

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

GitOrigin-RevId: 7c35662ddd
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
f79ed946e3
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      src/cambricon/impl/cambricon_runtime_opr.cpp

+ 6
- 0
src/cambricon/impl/cambricon_runtime_opr.cpp View File

@@ -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.",


Loading…
Cancel
Save