Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
1ce78aa0
MegEngine
项目概览
MegEngine 天元
/
MegEngine
11 个月 前同步成功
通知
392
Star
4702
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,发现更多精彩内容 >>
提交
1ce78aa0
编写于
2月 16, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(imperative): destruct dnn handles at last
GitOrigin-RevId: 7a67c68c55f56ef997758319d5c047c822c89e8b
上级
0cb60d64
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
9 addition
and
1 deletion
+9
-1
dnn/src/common/opr_delegate.h
dnn/src/common/opr_delegate.h
+2
-1
imperative/python/src/module.cpp
imperative/python/src/module.cpp
+7
-0
未找到文件。
dnn/src/common/opr_delegate.h
浏览文件 @
1ce78aa0
...
...
@@ -24,7 +24,8 @@ namespace megdnn {
* Usually used for calling other opr impls from some opr impl. You probably
* want to use CpuOprDelegationStorage instead.
*/
const
std
::
shared_ptr
<
Handle
>&
inplace_cpu_handle
(
int
debug_level
=
0
);
MGE_WIN_DECLSPEC_FUC
const
std
::
shared_ptr
<
Handle
>&
inplace_cpu_handle
(
int
debug_level
=
0
);
/*!
* \brief storage for oprs on inplace CPU handle
...
...
imperative/python/src/module.cpp
浏览文件 @
1ce78aa0
...
...
@@ -14,6 +14,7 @@
#define DO_IMPORT_ARRAY
#include "./helper.h"
#include "./numpy_dtypes.h"
#include "megdnn/handle.h"
#include "./common.h"
#include "./graph_rt.h"
...
...
@@ -30,6 +31,10 @@ using namespace mgb::imperative::python;
#define MODULE_NAME imperative_rt
#endif
namespace
megdnn
{
extern
const
std
::
shared_ptr
<
Handle
>&
inplace_cpu_handle
(
int
debug_level
=
0
);
}
PYBIND11_MODULE
(
MODULE_NAME
,
m
)
{
// initialize numpy
if
([]()
{
...
...
@@ -39,6 +44,8 @@ PYBIND11_MODULE(MODULE_NAME, m) {
throw
py
::
error_already_set
();
}
megdnn
::
inplace_cpu_handle
();
py
::
module
::
import
(
"sys"
).
attr
(
"modules"
)[
m
.
attr
(
"__name__"
)]
=
m
;
m
.
attr
(
"__package__"
)
=
m
.
attr
(
"__name__"
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录