Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
1bc09676
Mace
项目概览
Xiaomi
/
Mace
通知
106
Star
40
Fork
27
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
1bc09676
编写于
10月 15, 2020
作者:
Z
Zhang Zhimin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix micro matmul bias access bug
上级
6d16b491
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
11 addition
and
5 deletion
+11
-5
micro/ops/matmul.cc
micro/ops/matmul.cc
+7
-1
micro/ops/utils/gemv.cc
micro/ops/utils/gemv.cc
+4
-4
未找到文件。
micro/ops/matmul.cc
浏览文件 @
1bc09676
...
@@ -27,9 +27,15 @@ MaceStatus MatMulOp::OnInit() {
...
@@ -27,9 +27,15 @@ MaceStatus MatMulOp::OnInit() {
transpose_b_
=
GetArgByName
(
"transpose_b"
,
false
);
transpose_b_
=
GetArgByName
(
"transpose_b"
,
false
);
input_a_
=
GetInputData
<
mifloat
>
(
INPUT_A
);
input_a_
=
GetInputData
<
mifloat
>
(
INPUT_A
);
input_b_
=
GetInputData
<
mifloat
>
(
INPUT_B
);
input_b_
=
GetInputData
<
mifloat
>
(
INPUT_B
);
bias_
=
GetInputSize
()
>
3
?
GetInputData
<
mifloat
>
(
BIAS
)
:
NULL
;
output_
=
GetOutputData
<
mifloat
>
(
OUTPUT
);
output_
=
GetOutputData
<
mifloat
>
(
OUTPUT
);
bias_
=
NULL
;
if
(
GetInputSize
()
>=
3
)
{
bias_
=
GetInputData
<
mifloat
>
(
BIAS
);
bias_dim_size_
=
GetInputShapeDimSize
(
BIAS
);
bias_dims_
=
GetInputShapeDims
(
BIAS
);
}
input_a_dim_size_
=
GetInputShapeDimSize
(
INPUT_A
);
input_a_dim_size_
=
GetInputShapeDimSize
(
INPUT_A
);
input_b_dim_size_
=
GetInputShapeDimSize
(
INPUT_B
);
input_b_dim_size_
=
GetInputShapeDimSize
(
INPUT_B
);
...
...
micro/ops/utils/gemv.cc
浏览文件 @
1bc09676
...
@@ -98,10 +98,10 @@ MaceStatus Gemv<mifloat>::Compute(const mifloat *lhs_data,
...
@@ -98,10 +98,10 @@ MaceStatus Gemv<mifloat>::Compute(const mifloat *lhs_data,
float
sum2
=
0
;
float
sum2
=
0
;
float
sum3
=
0
;
float
sum3
=
0
;
if
(
bias_data
!=
NULL
)
{
if
(
bias_data
!=
NULL
)
{
sum0
=
bias_data
[
0
];
sum0
=
bias_data
[
h
+
0
];
sum1
=
bias_data
[
1
];
sum1
=
bias_data
[
h
+
1
];
sum2
=
bias_data
[
2
];
sum2
=
bias_data
[
h
+
2
];
sum3
=
bias_data
[
3
];
sum3
=
bias_data
[
h
+
3
];
}
}
const
int32_t
lhs_h_base0
=
(
lhs_b_base
+
h
)
*
lhs_width
;
const
int32_t
lhs_h_base0
=
(
lhs_b_base
+
h
)
*
lhs_width
;
const
int32_t
lhs_h_base1
=
lhs_h_base0
+
lhs_width
;
const
int32_t
lhs_h_base1
=
lhs_h_base0
+
lhs_width
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录