This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
0b4918b25e
commit
f79ed946e3
1 changed files
with
6 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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.",
Write
Preview
Loading…
Cancel
Save