Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
马赛克医院
Paddle
提交
54d3b5a1
P
Paddle
项目概览
马赛克医院
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
54d3b5a1
编写于
4月 11, 2020
作者:
Z
Zeng Jinle
提交者:
GitHub
4月 11, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
enhance err_msg, test=develop (#23714)
上级
a63bcf9a
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
31 addition
and
4 deletion
+31
-4
paddle/fluid/imperative/op_base.h
paddle/fluid/imperative/op_base.h
+22
-4
paddle/fluid/imperative/tests/test_layer.cc
paddle/fluid/imperative/tests/test_layer.cc
+9
-0
未找到文件。
paddle/fluid/imperative/op_base.h
浏览文件 @
54d3b5a1
...
@@ -42,13 +42,25 @@ class OpBase {
...
@@ -42,13 +42,25 @@ class OpBase {
~
OpBase
()
{
VLOG
(
3
)
<<
"Destruct Op: "
<<
Type
();
}
~
OpBase
()
{
VLOG
(
3
)
<<
"Destruct Op: "
<<
Type
();
}
const
std
::
string
&
Type
()
const
{
return
op_
->
Type
();
}
const
std
::
string
&
Type
()
const
{
return
op_
?
op_
->
Type
()
:
UnknownOpType
();
}
const
framework
::
AttributeMap
&
Attrs
()
const
{
return
attrs_
;
}
const
framework
::
AttributeMap
&
Attrs
()
const
{
return
attrs_
;
}
const
framework
::
OpInfo
&
Info
()
const
{
return
op_
->
Info
();
}
const
framework
::
OpInfo
&
Info
()
const
{
PADDLE_ENFORCE_NOT_NULL
(
op_
,
platform
::
errors
::
PreconditionNotMet
(
"OpBase::Info() should be called after "
"OpBase::SetType() is called"
));
return
op_
->
Info
();
}
const
framework
::
OperatorBase
&
InnerOp
()
const
{
return
*
op_
;
}
const
framework
::
OperatorBase
&
InnerOp
()
const
{
PADDLE_ENFORCE_NOT_NULL
(
op_
,
platform
::
errors
::
PreconditionNotMet
(
"OpBase::InnerOp() should be called after "
"OpBase::SetType() is called"
));
return
*
op_
;
}
void
ClearBackwardTrace
();
void
ClearBackwardTrace
();
...
@@ -63,7 +75,7 @@ class OpBase {
...
@@ -63,7 +75,7 @@ class OpBase {
void
SetType
(
const
std
::
string
&
type
);
void
SetType
(
const
std
::
string
&
type
);
void
CheckAttrs
()
{
void
CheckAttrs
()
{
auto
&
info
=
op_
->
Info
();
auto
&
info
=
Info
();
if
(
info
.
Checker
()
!=
nullptr
)
{
if
(
info
.
Checker
()
!=
nullptr
)
{
info
.
Checker
()
->
Check
(
&
attrs_
,
true
);
info
.
Checker
()
->
Check
(
&
attrs_
,
true
);
}
}
...
@@ -150,6 +162,12 @@ class OpBase {
...
@@ -150,6 +162,12 @@ class OpBase {
const
framework
::
AttributeMap
&
attrs
,
const
framework
::
AttributeMap
&
attrs
,
const
platform
::
Place
&
place
);
const
platform
::
Place
&
place
);
private:
static
const
std
::
string
&
UnknownOpType
()
{
static
std
::
string
kUnknownOpType
{
"unknown"
};
return
kUnknownOpType
;
}
private:
private:
NameVarMap
<
VariableWrapper
>
ins_
;
NameVarMap
<
VariableWrapper
>
ins_
;
NameVarMap
<
VariableWrapper
>
outs_
;
NameVarMap
<
VariableWrapper
>
outs_
;
...
...
paddle/fluid/imperative/tests/test_layer.cc
浏览文件 @
54d3b5a1
...
@@ -271,6 +271,15 @@ TEST(test_layer, test_dygraph_infershape_context) {
...
@@ -271,6 +271,15 @@ TEST(test_layer, test_dygraph_infershape_context) {
ASSERT_EQ
(
have_z
,
false
);
ASSERT_EQ
(
have_z
,
false
);
}
}
TEST
(
test_layer
,
test_inner_op_not_inited
)
{
OpBase
op
;
std
::
string
kUnknown
=
"unknown"
;
ASSERT_EQ
(
op
.
Type
(),
kUnknown
);
ASSERT_THROW
(
op
.
Info
(),
platform
::
EnforceNotMet
);
ASSERT_THROW
(
op
.
InnerOp
(),
platform
::
EnforceNotMet
);
ASSERT_THROW
(
op
.
CheckAttrs
(),
platform
::
EnforceNotMet
);
}
}
// namespace imperative
}
// namespace imperative
}
// namespace paddle
}
// namespace paddle
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录