Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
9330929f
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,发现更多精彩内容 >>
提交
9330929f
编写于
6月 30, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(lite): add c_opr_init_interface for lar
GitOrigin-RevId: f7bcc2765647a32951a197d4eb618c95e5fb735b
上级
16ba05a8
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
12 addition
and
4 deletion
+12
-4
lite/load_and_run/src/options/extern_c_opr_options.cpp
lite/load_and_run/src/options/extern_c_opr_options.cpp
+10
-3
lite/load_and_run/src/options/extern_c_opr_options.h
lite/load_and_run/src/options/extern_c_opr_options.h
+2
-1
未找到文件。
lite/load_and_run/src/options/extern_c_opr_options.cpp
浏览文件 @
9330929f
...
...
@@ -121,9 +121,13 @@ void COprLibOption::load_lib() {
handle
,
"failed to open c opr lib %s:
\n
errmsg: %s"
,
lib_path
.
c_str
(),
dlerror
());
const
char
*
entry
=
MGB_C_OPR_INIT_FUNC_STR
;
const
char
*
entry
=
m_c_opr_init_func
.
c_str
()
;
auto
func
=
dlsym
(
handle
,
entry
);
mgb_assert
(
func
,
"can not resolve %s: %s"
,
entry
,
dlerror
());
mgb_assert
(
func
,
"can not resolve %s: %s, please use '--c-opr-init-interface' to set the "
"init API of your loader"
,
entry
,
dlerror
());
typedef
void
(
*
entry_f_t
)(
void
*
);
reinterpret_cast
<
entry_f_t
>
(
func
)(
reinterpret_cast
<
void
*>
(
&
mgb_get_extern_c_opr_api_versioned
));
...
...
@@ -177,6 +181,7 @@ COprLibOption::COprLibOption() {
lib_path
=
FLAGS_c_opr_lib
;
c_opr_args
.
is_run_c_opr
=
!
lib_path
.
empty
();
c_opr_args
.
is_run_c_opr_with_param
=
FLAGS_c_opr_lib_with_param
;
m_c_opr_init_func
=
FLAGS_c_opr_init_interface
;
}
bool
COprLibOption
::
is_valid
()
{
...
...
@@ -205,5 +210,7 @@ DEFINE_bool(
"Run c opr lib with param, use to benchmark speed and check result, "
"need c opr loader implemente `copr_param_device_ptr_malloc, "
"copr_param_device_ptr_free and copr_param_device_ptr_h2d' symbols"
);
DEFINE_string
(
c_opr_init_interface
,
MGB_C_OPR_INIT_FUNC_STR
,
"set the C_OPR_INIT_FUNC interface when running"
);
REGIST_OPTION_CREATOR
(
c_opr_lib
,
lar
::
COprLibOption
::
create_option
);
lite/load_and_run/src/options/extern_c_opr_options.h
浏览文件 @
9330929f
...
...
@@ -6,7 +6,7 @@
DECLARE_bool
(
c_opr_lib_with_param
);
DECLARE_string
(
c_opr_lib
);
DECLARE_string
(
c_opr_init_interface
);
namespace
lar
{
struct
COprArgs
{
...
...
@@ -51,5 +51,6 @@ private:
COprArgs
c_opr_args
;
std
::
string
lib_path
;
std
::
shared_ptr
<
ExternCOprParam
>
c_opr_param
;
std
::
string
m_c_opr_init_func
;
};
}
// namespace lar
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录