Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
ea70d99b
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
396
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
ea70d99b
编写于
6月 29, 2021
作者:
M
Megvii Engine Team
提交者:
huangxinda
7月 19, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mge/convbias): make fallback convbias support nhwcd4 layout
GitOrigin-RevId: 1c306f867dc636b8e8349d4d42a17b914f160552
上级
497ef6c3
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
12 addition
and
14 deletion
+12
-14
dnn/src/common/relayout_format.cpp
dnn/src/common/relayout_format.cpp
+5
-13
dnn/src/fallback/conv_bias/opr_impl.cpp
dnn/src/fallback/conv_bias/opr_impl.cpp
+7
-1
未找到文件。
dnn/src/common/relayout_format.cpp
浏览文件 @
ea70d99b
...
...
@@ -382,19 +382,11 @@ void RelayoutFormat::deduce_format(TensorFormat src, TensorFormat& dst) {
if
(
dst
.
type
()
==
TensorFormat
::
Type
::
IMAGE2D_PACK4
&&
(
handle
()
->
type
()
!=
Handle
::
HandleType
::
NAIVE
))
{
#if MEGDNN_ENABLE_MANGLING
handle
()
->
type
()
!=
Handle
::
HandleType
::
NAIVE
&&
handle
()
->
type
()
!=
Handle
::
HandleType
::
X86
))
{
megdnn_throw
(
"Only naive and opencl handle support "
"Image2DPack4TensorFormat, try build with debug for get more "
"info"
);
#else
megdnn_throw
(
"Only naive and opencl handle support "
"Image2DPack4TensorFormat, try to export MGB_USE_MEGDNN_DBG=2 "
"and also export CUDA_VISIBLE_DEVICES=
\'\'
at CUDA env"
"to enable naive handle"
);
#endif
"Dump with Image2DPack4TensorFormat is not available on CUDA compnode, "
"try export CUDA_VISIBLE_DEVICES=
\'\'
"
);
}
#undef CHECK_SRC
}
...
...
dnn/src/fallback/conv_bias/opr_impl.cpp
浏览文件 @
ea70d99b
...
...
@@ -297,6 +297,9 @@ ConvBiasImpl::Algorithm* ConvBiasImpl::get_algorithm_heuristic_with_ncb(
const
NCBKernSizeParam
&
param
,
size_t
workspace_limit_in_bytes
,
const
AlgoAttribute
&
positive_attr
,
const
AlgoAttribute
&
negative_attr
)
{
if
(
ConvBiasImpl
::
param
().
format
==
Param
::
Format
::
NHWCD4
)
{
return
nullptr
;
}
auto
algo_data_type
=
param
.
deduce_algo_data_type
();
auto
suggest_category_order
=
suggest_algo_category_order
(
param
);
for
(
auto
category
:
suggest_category_order
)
{
...
...
@@ -346,7 +349,7 @@ ConvBiasImpl::NCBKernSizeParam ConvBiasImpl::make_ncb_kern_size_param(
param
().
format
==
Param
::
Format
::
NCHW32
||
param
().
format
==
Param
::
Format
::
NCHW64
)
{
spatial_pos
=
2
;
}
else
if
(
param
().
format
==
Param
::
Format
::
NHWC
)
{
}
else
if
(
param
().
format
==
Param
::
Format
::
NHWC
||
param
().
format
==
Param
::
Format
::
NHWCD4
)
{
spatial_pos
=
1
;
}
else
{
megdnn_assert
(
0
,
"invalid conv format %d"
,
...
...
@@ -497,6 +500,9 @@ ConvBiasImpl::Algorithm* ConvBiasImpl::get_algorithm_from_desc(
ConvBiasImpl
::
Algorithm
*
ConvBiasImpl
::
get_algorithm
(
const
NCBKernSizeParam
&
param
,
size_t
workspace_size
)
{
if
(
ConvBiasImpl
::
param
().
format
==
Param
::
Format
::
NHWCD4
)
{
return
nullptr
;
}
if
(
auto
algo
=
get_algorithm_from_desc
(
execution_policy
().
algo
))
{
return
algo
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录