Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
485e56ca
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,发现更多精彩内容 >>
提交
485e56ca
编写于
2月 02, 2023
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(lite): fix lite get model io info name error
GitOrigin-RevId: 904d86831d1a9dc6ebefb1c59463976e7667a339
上级
a450d0f5
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
28 addition
and
19 deletion
+28
-19
lite/lite-c/src/network.cpp
lite/lite-c/src/network.cpp
+12
-19
lite/pylite/test/test_global.py
lite/pylite/test/test_global.py
+16
-0
未找到文件。
lite/lite-c/src/network.cpp
浏览文件 @
485e56ca
...
...
@@ -170,22 +170,24 @@ struct InnerIO {
};
InnerIO
convert_to_inner_io
(
const
lite
::
NetworkIO
&
network_io
)
{
InnerIO
innner_io
;
for
(
size_t
i
=
0
;
i
<
network_io
.
inputs
.
size
();
i
++
)
{
InnerIO
inner_io
;
size_t
id
=
0
;
inner_io
.
names
.
resize
(
network_io
.
inputs
.
size
()
+
network_io
.
outputs
.
size
());
for
(
size_t
i
=
0
;
i
<
network_io
.
inputs
.
size
();
i
++
,
id
++
)
{
lite
::
IO
io
=
network_io
.
inputs
[
i
];
inn
ner_io
.
names
.
push_back
(
io
.
name
)
;
inn
n
er_io
.
inputs
.
push_back
(
{
inn
ner_io
.
names
.
back
()
.
c_str
(),
io
.
is_host
,
io
.
io_type
,
inn
er_io
.
names
[
id
]
=
io
.
name
;
inner_io
.
inputs
.
push_back
(
{
inn
er_io
.
names
[
id
]
.
c_str
(),
io
.
is_host
,
io
.
io_type
,
convert_to_clayout
(
io
.
config_layout
)});
}
for
(
size_t
i
=
0
;
i
<
network_io
.
outputs
.
size
();
i
++
)
{
for
(
size_t
i
=
0
;
i
<
network_io
.
outputs
.
size
();
i
++
,
id
++
)
{
lite
::
IO
io
=
network_io
.
outputs
[
i
];
inn
ner_io
.
names
.
push_back
(
io
.
name
)
;
inn
n
er_io
.
outputs
.
push_back
(
{
inn
ner_io
.
names
.
back
()
.
c_str
(),
io
.
is_host
,
io
.
io_type
,
inn
er_io
.
names
[
id
]
=
io
.
name
;
inner_io
.
outputs
.
push_back
(
{
inn
er_io
.
names
[
id
]
.
c_str
(),
io
.
is_host
,
io
.
io_type
,
convert_to_clayout
(
io
.
config_layout
)});
}
return
inn
n
er_io
;
return
inner_io
;
}
lite
::
ExtraConfig
convert_extra_config
(
const
LiteExtraConfig
&
extra_config
)
{
...
...
@@ -727,15 +729,6 @@ int write_ios_from_cpp_io(
ios
->
output_size
=
inner_io
.
outputs
.
size
();
ios
->
inputs
=
inner_io
.
inputs
.
data
();
ios
->
outputs
=
inner_io
.
outputs
.
data
();
size_t
i
=
0
;
for
(;
i
<
ios
->
input_size
;
i
++
)
{
auto
io_ptr
=
ios
->
inputs
+
i
;
io_ptr
->
name
=
inner_io
.
names
[
i
].
c_str
();
}
for
(;
i
<
ios
->
output_size
;
i
++
)
{
auto
io_ptr
=
ios
->
outputs
+
i
;
io_ptr
->
name
=
inner_io
.
names
[
i
].
c_str
();
}
LITE_CAPI_END
();
}
...
...
lite/pylite/test/test_global.py
浏览文件 @
485e56ca
...
...
@@ -78,3 +78,19 @@ class TestGlobal(TestShuffleNet):
phy_ptr2
=
LiteGlobal
.
lookup_physic_ptr
(
vir_ptr
,
LiteDeviceType
.
LITE_AX
)
assert
phy_ptr
.
value
==
phy_ptr2
.
value
LiteGlobal
.
clear_memory_pair
(
vir_ptr
,
phy_ptr
,
LiteDeviceType
.
LITE_AX
)
def
test_get_model_io_info
():
source_dir
=
os
.
getenv
(
"LITE_TEST_RESOURCE"
)
model_path
=
os
.
path
.
join
(
source_dir
,
"./ax_models/77-fcf1a1af.axe"
)
model_io
=
get_model_io_info
(
model_path
)
input_names
=
[
in_node
.
name
for
in_node
in
model_io
.
inputs
]
output_names
=
[
out_node
.
name
for
out_node
in
model_io
.
outputs
]
assert
"op_4121.hardware"
in
input_names
assert
"op_4238.hardware"
in
input_names
assert
"op_4218:u2s.hardware"
in
input_names
assert
"op_5034:add"
in
output_names
assert
"op_5035:add"
in
output_names
assert
"op_5036:add"
in
output_names
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录