Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
1af48631
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
403
Star
4705
Fork
582
代码
文件
提交
分支
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看板
提交
1af48631
编写于
5月 11, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(dnn/fallback): fix conv1x1's is_preferred method
GitOrigin-RevId: 6d34080d2557614cea97443d7a8153299714be31
上级
206521cd
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
9 addition
and
3 deletion
+9
-3
dnn/src/aarch64/matrix_mul/opr_impl.cpp
dnn/src/aarch64/matrix_mul/opr_impl.cpp
+1
-1
dnn/src/armv7/matrix_mul/opr_impl.cpp
dnn/src/armv7/matrix_mul/opr_impl.cpp
+2
-2
dnn/src/fallback/conv_bias/conv1x1/algos.h
dnn/src/fallback/conv_bias/conv1x1/algos.h
+4
-0
src/core/impl/comp_node/cpu/comp_node.cpp
src/core/impl/comp_node/cpu/comp_node.cpp
+2
-0
未找到文件。
dnn/src/aarch64/matrix_mul/opr_impl.cpp
浏览文件 @
1af48631
...
...
@@ -64,8 +64,8 @@ public:
all_algos
.
emplace_back
(
&
int8x8x32_k8x12x4_dotprod
);
#else
all_algos
.
emplace_back
(
&
int8x8x32_gemv
);
all_algos
.
emplace_back
(
&
int8x8x32_k8x8x8
);
all_algos
.
emplace_back
(
&
int8x8x32_k4x4x16
);
all_algos
.
emplace_back
(
&
int8x8x32_k8x8x8
);
all_algos
.
emplace_back
(
&
int8x8x32_mk4_4x4x16
);
#endif
all_algos
.
emplace_back
(
&
int8x8x16_k4x4x16
);
...
...
dnn/src/armv7/matrix_mul/opr_impl.cpp
浏览文件 @
1af48631
...
...
@@ -61,11 +61,11 @@ public:
all_algos
.
emplace_back
(
&
int8x8x32_gemv
);
#endif
all_algos
.
emplace_back
(
&
int8x8x32_mk4_4x2x16
);
all_algos
.
emplace_back
(
&
int8x8x32_k4x8x8
);
all_algos
.
emplace_back
(
&
int8x8x32_k4x2x16
);
all_algos
.
emplace_back
(
&
int8x8x32_k4x8x8
);
all_algos
.
emplace_back
(
&
quint8_k4x8x8
);
all_algos
.
emplace_back
(
&
int8x8x16_k4x8x8
);
all_algos
.
emplace_back
(
&
int8x8x16_k4x2x16
);
all_algos
.
emplace_back
(
&
int8x8x16_k4x8x8
);
all_algos
.
emplace_back
(
&
int16x16x32_k12x4x1
);
all_algos
.
emplace_back
(
&
int16x16x32_mk8_4x8
);
}
...
...
dnn/src/fallback/conv_bias/conv1x1/algos.h
浏览文件 @
1af48631
...
...
@@ -41,6 +41,10 @@ public:
SmallVector
<
NCBKern
>
dispatch_kerns
(
ConvBiasImpl
*
opr
,
const
NCBKernSizeParam
&
param
)
const
override
;
bool
is_preferred
(
ConvBiasImpl
*
,
const
NCBKernSizeParam
&
)
const
override
{
return
true
;
}
protected:
size_t
get_oc_tile_size_heuristic
(
const
NCBKernSizeParam
&
param
)
const
;
...
...
src/core/impl/comp_node/cpu/comp_node.cpp
浏览文件 @
1af48631
...
...
@@ -58,7 +58,9 @@ class CpuCompNode::WorkerQueue final
void
on_async_queue_worker_thread_start
()
override
{
mgb_assert
(
m_locator
.
device
>=
0
);
if
(
enable_affinity
)
{
#if !defined(ANDROID) && !defined(__ANDROID__)
sys
::
set_cpu_affinity
({
m_locator
.
device
});
#endif
}
sys
::
set_thread_name
(
m_locator
.
to_string
());
if
(
m_thread_pool
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录