Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
1c9430e2
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看板
提交
1c9430e2
编写于
8月 31, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mge/imperative): clear log message
GitOrigin-RevId: 051264cac3eabfdb3d6ccc0a46586b9a4f258bb5
上级
a226f02e
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
10 addition
and
0 deletion
+10
-0
imperative/python/tools/gen_ops.py
imperative/python/tools/gen_ops.py
+8
-0
imperative/src/impl/op_trait.cpp
imperative/src/impl/op_trait.cpp
+2
-0
未找到文件。
imperative/python/tools/gen_ops.py
浏览文件 @
1c9430e2
...
@@ -52,6 +52,7 @@ class Context:
...
@@ -52,6 +52,7 @@ class Context:
self
.
indent
=
0
self
.
indent
=
0
self
.
generated
=
[]
self
.
generated
=
[]
self
.
skipped
=
[]
self
.
skipped
=
[]
self
.
generated_signature
=
set
()
def
write
(
self
,
text
,
*
fmt
,
indent
=
0
):
def
write
(
self
,
text
,
*
fmt
,
indent
=
0
):
text
=
textwrap
.
dedent
(
text
)
text
=
textwrap
.
dedent
(
text
)
...
@@ -184,6 +185,13 @@ class Context:
...
@@ -184,6 +185,13 @@ class Context:
self
.
skipped
.
append
(
name
)
self
.
skipped
.
append
(
name
)
return
return
signature
=
(
name
,
params
if
isinstance
(
params
,
str
)
else
frozenset
(
params
),
has_out_dtype
,
version
)
if
signature
in
self
.
generated_signature
:
self
.
skipped
.
append
(
name
)
return
else
:
self
.
generated_signature
.
add
(
signature
)
body
=
body
or
[]
body
=
body
or
[]
if
isinstance
(
params
,
str
):
if
isinstance
(
params
,
str
):
params
=
[(
'param'
,
params
)]
params
=
[(
'param'
,
params
)]
...
...
imperative/src/impl/op_trait.cpp
浏览文件 @
1c9430e2
...
@@ -87,12 +87,14 @@ OpTraitRegistry& OpTraitRegistry::finalize() {
...
@@ -87,12 +87,14 @@ OpTraitRegistry& OpTraitRegistry::finalize() {
CHECK
(
infer_output_attrs
);
CHECK
(
infer_output_attrs
);
CHECK
(
make_backward_graph
);
CHECK
(
make_backward_graph
);
#undef CHECK
#undef CHECK
#ifdef DEBUG
if
(
msg
.
tellp
()
>
0
)
{
if
(
msg
.
tellp
()
>
0
)
{
mgb_log_warn
(
mgb_log_warn
(
"%s op trait missing: %s"
,
"%s op trait missing: %s"
,
trait
->
name
?
trait
->
name
:
"(anonymous)"
,
trait
->
name
?
trait
->
name
:
"(anonymous)"
,
msg
.
str
().
c_str
()
+
2
/* skip first ", " */
);
msg
.
str
().
c_str
()
+
2
/* skip first ", " */
);
}
}
#endif
return
*
this
;
return
*
this
;
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录