Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle-Lite
提交
e6fe420e
P
Paddle-Lite
项目概览
PaddlePaddle
/
Paddle-Lite
通知
331
Star
4
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
271
列表
看板
标记
里程碑
合并请求
78
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle-Lite
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
271
Issue
271
列表
看板
标记
里程碑
合并请求
78
合并请求
78
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
e6fe420e
编写于
6月 20, 2018
作者:
R
Ruilong Liu
提交者:
GitHub
6月 20, 2018
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'develop' into develop
上级
53e70f91
5965b396
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
18 addition
and
5 deletion
+18
-5
src/operators/math/gemm.cpp
src/operators/math/gemm.cpp
+18
-5
未找到文件。
src/operators/math/gemm.cpp
浏览文件 @
e6fe420e
...
@@ -20,7 +20,9 @@ limitations under the License. */
...
@@ -20,7 +20,9 @@ limitations under the License. */
namespace
paddle_mobile
{
namespace
paddle_mobile
{
namespace
operators
{
namespace
operators
{
namespace
math
{
namespace
math
{
float
ab
[
MR
*
NR
];
alignas
(
64
)
float
packedA
[
MC
*
KC
];
alignas
(
64
)
float
packedB
[
KC
*
NC
];
alignas
(
64
)
float
ab
[
MR
*
NR
];
// 将A矩阵分块复制到连续内存(ColMajor)
// 将A矩阵分块复制到连续内存(ColMajor)
void
PackMatrixA
(
int
m
,
int
k
,
int
paddingM
,
const
float
*
A
,
int
lda
,
void
PackMatrixA
(
int
m
,
int
k
,
int
paddingM
,
const
float
*
A
,
int
lda
,
float
*
buffer
)
{
float
*
buffer
)
{
...
@@ -153,9 +155,6 @@ void InnerKernel(int m, int n, int k, float alpha, const float *A, int lda,
...
@@ -153,9 +155,6 @@ void InnerKernel(int m, int n, int k, float alpha, const float *A, int lda,
Buff_B_N
=
n
+
(
NR
-
_nc
);
Buff_B_N
=
n
+
(
NR
-
_nc
);
}
}
float
packedA
[
MC
*
KC
];
static
float
packedB
[
KC
*
NC
];
if
(
first_time
)
{
if
(
first_time
)
{
PackMatrixB_
(
k
,
n
,
_nc
,
B
,
ldb
,
packedB
);
PackMatrixB_
(
k
,
n
,
_nc
,
B
,
ldb
,
packedB
);
}
}
...
@@ -324,7 +323,7 @@ void AddDot4x4_relu(int k, float alpha, const float *a, int lda, const float *b,
...
@@ -324,7 +323,7 @@ void AddDot4x4_relu(int k, float alpha, const float *a, int lda, const float *b,
#elif defined(ARMV7)
#elif defined(ARMV7)
void
AddDot4x4
(
int
k
,
float
alpha
,
const
float
*
a
,
int
lda
,
const
float
*
b
,
void
AddDot4x4
(
int
k
,
float
alpha
,
const
float
*
a
,
int
lda
,
const
float
*
b
,
int
ldb
,
float
beta
,
float
*
C
,
int
ldc
,
int
mc
,
int
nc
)
{
int
ldb
,
float
beta
,
float
*
C
,
int
ldc
,
int
mc
,
int
nc
)
{
int
kc1
=
k
/
2
,
kc2
=
k
%
2
;
int
kc1
=
k
/
4
,
kc2
=
k
%
4
;
int
bytes_ldc
=
4
*
ldc
;
int
bytes_ldc
=
4
*
ldc
;
int
flag_alpha
=
(
alpha
==
1.0
)
?
1
:
2
;
int
flag_alpha
=
(
alpha
==
1.0
)
?
1
:
2
;
int
flag_beta
;
int
flag_beta
;
...
@@ -336,6 +335,8 @@ void AddDot4x4(int k, float alpha, const float *a, int lda, const float *b,
...
@@ -336,6 +335,8 @@ void AddDot4x4(int k, float alpha, const float *a, int lda, const float *b,
flag_beta
=
2
;
flag_beta
=
2
;
}
}
asm
volatile
(
asm
volatile
(
"pld [%[a]]
\n\t
"
"pld [%[b]]
\n\t
"
"vmov.f32 q10, #0.0
\n\t
"
"vmov.f32 q10, #0.0
\n\t
"
"vmov.f32 q11, #0.0
\n\t
"
"vmov.f32 q11, #0.0
\n\t
"
"vmov.f32 q12, #0.0
\n\t
"
"vmov.f32 q12, #0.0
\n\t
"
...
@@ -344,6 +345,18 @@ void AddDot4x4(int k, float alpha, const float *a, int lda, const float *b,
...
@@ -344,6 +345,18 @@ void AddDot4x4(int k, float alpha, const float *a, int lda, const float *b,
"subs %[kc1], %[kc1], #1
\n\t
"
"subs %[kc1], %[kc1], #1
\n\t
"
"blt end_kc1_%=
\n\t
"
"blt end_kc1_%=
\n\t
"
"loop_kc1_%=:
\n\t
"
"loop_kc1_%=:
\n\t
"
"pld [%[a], #64]
\n\t
"
"pld [%[b], #64]
\n\t
"
"vld1.32 {q0, q1}, [%[a]]!
\n\t
"
"vld1.32 {q2, q3}, [%[b]]!
\n\t
"
"vmla.f32 q10, q2, d0[0]
\n\t
"
"vmla.f32 q11, q2, d0[1]
\n\t
"
"vmla.f32 q12, q2, d1[0]
\n\t
"
"vmla.f32 q13, q2, d1[1]
\n\t
"
"vmla.f32 q10, q3, d2[0]
\n\t
"
"vmla.f32 q11, q3, d2[1]
\n\t
"
"vmla.f32 q12, q3, d3[0]
\n\t
"
"vmla.f32 q13, q3, d3[1]
\n\t
"
"vld1.32 {q0, q1}, [%[a]]!
\n\t
"
"vld1.32 {q0, q1}, [%[a]]!
\n\t
"
"vld1.32 {q2, q3}, [%[b]]!
\n\t
"
"vld1.32 {q2, q3}, [%[b]]!
\n\t
"
"vmla.f32 q10, q2, d0[0]
\n\t
"
"vmla.f32 q10, q2, d0[0]
\n\t
"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录