Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
20ab82d0
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
407
Star
4707
Fork
583
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
20ab82d0
编写于
3年前
作者:
M
Megvii Engine Team
提交者:
huangxinda
3年前
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(tee): fix tee crash
GitOrigin-RevId: 379f970c8777f6ba1e6acce5ab46ff1dc6954a1c
上级
933dd9a4
master
HuaHua404-patch-1
HuaHua404-patch-2
HuaHua404-patch-3
HuaHua404-patch-4
add-ci
add-dingding-notification
add-tools
dev-support-lite-fork-debug-mode
docstring-reshape
release-1.10
release-1.11
release-1.11.1
release-1.12.0
release-1.12.1
release-1.12.2
release-1.12.3
release-1.12.4
release-1.12.5
release-1.13.0
release-1.13.1
release-1.5
release-1.6
release-1.7
release-1.8
release-1.9
revert-211-master
revert-410-docstring-zeros
revert-411-add-tools
test-try-import
tmp-test
try-import
v1.13.1
v1.13.0
v1.12.4
v1.12.3
v1.12.2
v1.12.1
v1.12.0
v1.11.1
v1.11.0
v1.10.0
v1.9.1
v1.9.0
v1.8.2
v1.8.1
v1.8.1.m1
v1.8.0
v1.7.2.m1
v1.7.1.m1
v1.7.0
v1.7.0.m1
v1.6.0
v1.6.0-rc1
v1.5.0
无相关合并请求
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
16 addition
and
0 deletion
+16
-0
dnn/src/aarch64/matrix_mul/algos.cpp
dnn/src/aarch64/matrix_mul/algos.cpp
+6
-0
dnn/src/aarch64/matrix_mul/fp32/strategy.cpp
dnn/src/aarch64/matrix_mul/fp32/strategy.cpp
+6
-0
dnn/src/arm_common/conv_bias/fp32/direct_kernels/f32_direct_nchw_nchw44_kern_common.h
.../fp32/direct_kernels/f32_direct_nchw_nchw44_kern_common.h
+4
-0
未找到文件。
dnn/src/aarch64/matrix_mul/algos.cpp
浏览文件 @
20ab82d0
...
...
@@ -908,6 +908,9 @@ bool MatrixMulImpl::AlgoInt8x8x16MK4_16x12x4::preferred(
return
false
;
#else
auto
arch
=
cpuinfo_get_current_core
()
->
uarch
;
#ifdef __IN_TEE_ENV__
arch
=
cpuinfo_uarch_unknown
;
#endif
bool
little_core
=
arch
==
cpuinfo_uarch_cortex_a53
||
arch
==
cpuinfo_uarch_cortex_a55
;
return
little_core
;
...
...
@@ -986,6 +989,9 @@ bool MatrixMulImpl::AlgoInt8x8x16MK4_4x4x8::preferred(
return
false
;
#else
auto
arch
=
cpuinfo_get_current_core
()
->
uarch
;
#ifdef __IN_TEE_ENV__
arch
=
cpuinfo_uarch_unknown
;
#endif
bool
little_core
=
arch
==
cpuinfo_uarch_cortex_a53
||
arch
==
cpuinfo_uarch_cortex_a55
;
return
!
little_core
;
...
...
This diff is collapsed.
Click to expand it.
dnn/src/aarch64/matrix_mul/fp32/strategy.cpp
浏览文件 @
20ab82d0
...
...
@@ -181,6 +181,9 @@ void sgemm_8x12::kern(const float* packA, const float* packB, size_t M,
is_first_k
);
#else
auto
arch
=
cpuinfo_get_current_core
()
->
uarch
;
#ifdef __IN_TEE_ENV__
arch
=
cpuinfo_uarch_unknown
;
#endif
if
(
arch
==
cpuinfo_uarch_cortex_a53
)
{
sgemm_8x12_helper
<
matmul_general_8x12_a53
>
(
packA
,
packB
,
M
,
N
,
K
,
C
,
LDC
,
is_first_k
);
...
...
@@ -274,6 +277,9 @@ void sgemm_mk4_8x12::kern(const float* packA, const float* packB, size_t M,
is_first_k
);
#else
auto
arch
=
cpuinfo_get_current_core
()
->
uarch
;
#ifdef __IN_TEE_ENV__
arch
=
cpuinfo_uarch_unknown
;
#endif
if
(
arch
==
cpuinfo_uarch_cortex_a53
)
{
sgemm_mk4_8x12_helper
<
matmul_mk4_8x12_a53
>
(
packA
,
packB
,
M
,
N
,
K
,
C
,
LDC
,
is_first_k
);
...
...
This diff is collapsed.
Click to expand it.
dnn/src/arm_common/conv_bias/fp32/direct_kernels/f32_direct_nchw_nchw44_kern_common.h
浏览文件 @
20ab82d0
...
...
@@ -851,6 +851,10 @@ struct ConvDirectFp32NchwNchw44<bias_mode, Op, 3, 2> {
#else
auto
arch_tag
=
CpuTag
::
A7_TAG
;
#endif
#ifdef __IN_TEE_ENV__
arch_tag
=
CpuTag
::
DEFAULT_CPU_TAG
;
#endif
if
(
arch_tag
==
CpuTag
::
A7_TAG
)
{
for
(
int
oc_idx
=
0
;
oc_idx
<
oc_end
;
oc_idx
+=
big_oc_step
)
{
const
int
weight_offset
=
oc_idx
*
ic
*
fh
*
fw
;
...
...
This diff is collapsed.
Click to expand it.
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录
反馈
建议
客服
返回
顶部