Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
c3d5b61f
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
403
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看板
提交
c3d5b61f
编写于
4月 30, 2020
作者:
M
Megvii Engine Team
提交者:
Xinran Xu
5月 12, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(mgb/opr): add property params for python operator class
GitOrigin-RevId: af6da0e0ac18fa8d81c2415dfe982ae6ce71451a
上级
81915f6b
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
25 addition
and
1 deletion
+25
-1
python_module/src/cpp/megbrain_wrap.cpp
python_module/src/cpp/megbrain_wrap.cpp
+5
-0
python_module/src/cpp/megbrain_wrap.h
python_module/src/cpp/megbrain_wrap.h
+11
-1
python_module/src/swig/operator.i
python_module/src/swig/operator.i
+4
-0
python_module/src/swig/operator.py
python_module/src/swig/operator.py
+5
-0
未找到文件。
python_module/src/cpp/megbrain_wrap.cpp
浏览文件 @
c3d5b61f
...
...
@@ -880,6 +880,11 @@ SymbolVar SharedScalar::_as_sym_var(CompGraph &cg, mgb::CompNode &cn) {
ssprintf
(
"SharedScalar@%p"
,
m_val
.
get
()));
}
/* =============== Operator =============== */
const
std
::
unique_ptr
<
mgb
::
OprFootprint
>
Operator
::
sm_opr_footprint_ptr
{
std
::
make_unique
<
mgb
::
OprFootprint
>
()};
/* ================= misc ================= */
SymbolVar
fill_retain_dtype
(
SymbolVar
var
,
PyObject
*
value
)
{
...
...
python_module/src/cpp/megbrain_wrap.h
浏览文件 @
c3d5b61f
...
...
@@ -17,6 +17,8 @@
#include "megbrain/graph.h"
#include "megbrain/opr/io.h"
#include "megbrain/plugin/opr_footprint.h"
#include <map>
#include <string>
...
...
@@ -441,16 +443,24 @@ class SharedScalar {
*/
class
Operator
{
mgb
::
cg
::
OperatorNodeBase
*
m_operator_node
;
std
::
string
m_params
;
static
const
std
::
unique_ptr
<
mgb
::
OprFootprint
>
sm_opr_footprint_ptr
;
public:
Operator
()
:
m_operator_node
(
nullptr
){};
Operator
(
mgb
::
cg
::
OperatorNodeBase
*
operator_node
)
:
m_operator_node
(
operator_node
)
{}
:
m_operator_node
(
operator_node
),
m_params
(
std
::
move
(
(
sm_opr_footprint_ptr
->
calc_footprint
(
m_operator_node
)).
param
->
to_string
()))
{}
size_t
id
()
const
{
return
m_operator_node
->
id
();
}
const
std
::
string
&
name
()
const
{
return
m_operator_node
->
name
();
}
const
std
::
string
&
params
()
const
{
return
m_params
;
}
const
std
::
shared_ptr
<
mgb
::
ComputingGraph
>
get_owner_graph
()
const
{
return
m_operator_node
->
owner_graph
()
->
shared_from_this
();
}
...
...
python_module/src/swig/operator.i
浏览文件 @
c3d5b61f
...
...
@@ -58,6 +58,10 @@ public:
return
$
self-
>
name
()
;
}
const
std
::
string&
_get_params
()
const
{
return
$
self-
>
params
()
;
}
SymbolVarArray
_get_inputs
()
{
return
$
self-
>
inputs
()
;
}
...
...
python_module/src/swig/operator.py
浏览文件 @
c3d5b61f
...
...
@@ -19,6 +19,11 @@ def id(self):
def
name
(
self
):
return
self
.
_get_name
()
@
property
def
params
(
self
):
import
json
return
json
.
loads
(
self
.
_get_params
())
@
property
def
inputs
(
self
):
return
tuple
(
self
.
_get_inputs
())
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录