Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
e9c036cc
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
399
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,发现更多精彩内容 >>
提交
e9c036cc
编写于
12月 19, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mge/imperative): fix op specializations and tuple hash
GitOrigin-RevId: e3df93fd5b1303aeab76019f4a75596a051a5bf1
上级
b5ec83c5
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
34 addition
and
21 deletion
+34
-21
imperative/src/impl/ops/autogen.cpp
imperative/src/impl/ops/autogen.cpp
+19
-14
imperative/src/impl/ops/specializations.cpp
imperative/src/impl/ops/specializations.cpp
+15
-7
未找到文件。
imperative/src/impl/ops/autogen.cpp
浏览文件 @
e9c036cc
...
...
@@ -18,23 +18,28 @@ using namespace megdnn;
// FIXME: remove this when mgb::hash support tuple_hash
namespace
mgb
{
namespace
{
template
<
typename
T
,
size_t
...
Ns
>
auto
tail
(
T
t
,
std
::
index_sequence
<
Ns
...
>
)
{
return
std
::
make_tuple
(
std
::
get
<
Ns
+
1
>
(
t
)...);
struct
HashWrapper
{
size_t
hash
;
constexpr
operator
size_t
()
{
return
hash
;}
constexpr
HashWrapper
operator
+
(
HashWrapper
rhs
)
{
// NOTE: use a + b + c + d, not a + (b + (c + d)) !!!
return
{
hash
*
20141203
+
rhs
.
hash
};
}
};
template
<
typename
...
Args
>
constexpr
size_t
hash_many
(
const
Args
&
...
args
)
{
return
(...
+
HashWrapper
{
mgb
::
hash
(
args
)});
}
}
// anonymous namespace
template
<
typename
T
,
typename
...
Args
>
class
HashTrait
<
std
::
tuple
<
T
,
Args
...
>>
{
constexpr
static
size_t
length
=
sizeof
...(
Args
);
public:
struct
HashTrait
<
std
::
tuple
<
T
,
Args
...
>>
{
static
size_t
eval
(
const
std
::
tuple
<
T
,
Args
...
>
&
t
)
{
const
T
&
val
=
std
::
get
<
0
>
(
t
);
if
constexpr
(
!
length
)
{
return
mgb
::
hash
(
val
);
}
else
{
return
mgb
::
hash_pair_combine
(
mgb
::
hash
(
val
),
mgb
::
hash
(
tail
(
t
,
std
::
make_index_sequence
<
length
-
1
>
{})));
}
return
std
::
apply
(
hash_many
<
T
,
Args
...
>
,
t
);
}
};
}
// namespace mgb
...
...
imperative/src/impl/ops/specializations.cpp
浏览文件 @
e9c036cc
...
...
@@ -158,7 +158,13 @@ auto apply_on_var_node(
}
}
OP_TRAIT_REG
(
Reduce
,
Reduce
)
std
::
shared_ptr
<
OpDef
>
make_from_op_node
(
cg
::
OperatorNodeBase
*
node_
)
{
auto
*
node
=
&
node_
->
cast_final_safe
<
opr
::
Reduce
>
();
return
Reduce
::
make
(
node
->
param
());
}
OP_TRAIT_REG
(
Reduce
,
Reduce
,
opr
::
Reduce
)
.
make_from_op_node
(
make_from_op_node
)
.
apply_on_var_node
(
apply_on_var_node
)
.
fallback
();
}}
// reduce
...
...
@@ -439,12 +445,13 @@ OP_TRAIT_REG(GaussianRNG, GaussianRNG)
}}
// gaussian_rng
namespace
{
namespace
roi_align
{
auto
apply_on_var_node
(
VarNodeArray
apply_on_var_node
(
const
OpDef
&
def
,
const
VarNodeArray
&
inputs
)
{
auto
&&
op
=
static_cast
<
const
ROIAlign
&>
(
def
);
mgb_assert
(
inputs
.
size
()
==
2
);
return
opr
::
ROIAlign
::
make
(
inputs
[
0
],
inputs
[
1
],
op
.
param
());
auto
*
opr
=
opr
::
ROIAlign
::
make
(
inputs
[
0
],
inputs
[
1
],
op
.
param
()).
node
()
->
owner_opr
();
return
{
opr
->
output
(
0
),
opr
->
output
(
1
)};
}
OP_TRAIT_REG
(
ROIAlign
,
ROIAlign
)
.
apply_on_var_node
(
apply_on_var_node
)
...
...
@@ -496,12 +503,13 @@ OP_TRAIT_REG(Eye, Eye)
}}
// eye
namespace
{
namespace
roi_pooling
{
auto
apply_on_var_node
(
VarNodeArray
apply_on_var_node
(
const
OpDef
&
def
,
const
VarNodeArray
&
inputs
)
{
auto
&&
op
=
static_cast
<
const
ROIPooling
&>
(
def
);
mgb_assert
(
inputs
.
size
()
==
3
);
return
opr
::
ROIPooling
::
make
(
inputs
[
0
],
inputs
[
1
],
inputs
[
2
],
op
.
param
());
auto
*
opr
=
opr
::
ROIPooling
::
make
(
inputs
[
0
],
inputs
[
1
],
inputs
[
2
],
op
.
param
()).
node
()
->
owner_opr
();
return
{
opr
->
output
(
0
),
opr
->
output
(
1
)};
}
OP_TRAIT_REG
(
ROIPooling
,
ROIPooling
)
.
apply_on_var_node
(
apply_on_var_node
)
...
...
@@ -620,7 +628,7 @@ auto apply_on_var_node(
const
VarNodeArray
&
inputs
)
{
auto
&&
op
=
static_cast
<
const
SVD
&>
(
def
);
mgb_assert
(
inputs
.
size
()
==
1
);
return
opr
::
SVD
::
make
(
inputs
[
0
],
op
.
param
());
return
opr
::
SVD
::
make
(
inputs
[
0
],
op
.
param
())
[
0
].
node
()
->
owner_opr
()
->
usable_output
()
;
}
OP_TRAIT_REG
(
SVD
,
SVD
)
.
apply_on_var_node
(
apply_on_var_node
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录