Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
8bd2879e
P
Paddle
项目概览
BaiXuePrincess
/
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看板
未验证
提交
8bd2879e
编写于
12月 18, 2020
作者:
石
石晓伟
提交者:
GitHub
12月 18, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update the operator registration for incompatible upgrade, test=develop (#29720)
上级
10edfb6f
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
79 addition
and
12 deletion
+79
-12
paddle/fluid/framework/op_version_registry.cc
paddle/fluid/framework/op_version_registry.cc
+31
-0
paddle/fluid/framework/op_version_registry.h
paddle/fluid/framework/op_version_registry.h
+35
-12
paddle/fluid/framework/op_version_registry_test.cc
paddle/fluid/framework/op_version_registry_test.cc
+13
-0
未找到文件。
paddle/fluid/framework/op_version_registry.cc
浏览文件 @
8bd2879e
...
...
@@ -62,6 +62,37 @@ OpVersionDesc&& OpVersionDesc::BugfixWithBehaviorChanged(
return
std
::
move
(
*
this
);
}
OpVersionDesc
&&
OpVersionDesc
::
DeleteAttr
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
)
{
infos_
.
emplace_back
(
new_update
<
OpUpdateType
::
kDeleteAttr
>
(
OpAttrInfo
(
name
,
remark
)));
return
std
::
move
(
*
this
);
}
OpVersionDesc
&&
OpVersionDesc
::
ModifyInput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
)
{
infos_
.
emplace_back
(
new_update
<
OpUpdateType
::
kModifyInput
>
(
OpInputOutputInfo
(
name
,
remark
)));
return
std
::
move
(
*
this
);
}
OpVersionDesc
&&
OpVersionDesc
::
ModifyOutput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
)
{
infos_
.
emplace_back
(
new_update
<
OpUpdateType
::
kModifyOutput
>
(
OpInputOutputInfo
(
name
,
remark
)));
return
std
::
move
(
*
this
);
}
OpVersionDesc
&&
OpVersionDesc
::
DeleteInput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
)
{
infos_
.
emplace_back
(
new_update
<
OpUpdateType
::
kDeleteInput
>
(
OpInputOutputInfo
(
name
,
remark
)));
return
std
::
move
(
*
this
);
}
OpVersionDesc
&&
OpVersionDesc
::
DeleteOutput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
)
{
infos_
.
emplace_back
(
new_update
<
OpUpdateType
::
kDeleteOutput
>
(
OpInputOutputInfo
(
name
,
remark
)));
return
std
::
move
(
*
this
);
}
OpVersion
&
OpVersionRegistrar
::
Register
(
const
std
::
string
&
op_type
)
{
PADDLE_ENFORCE_EQ
(
op_version_map_
.
find
(
op_type
),
op_version_map_
.
end
(),
...
...
paddle/fluid/framework/op_version_registry.h
浏览文件 @
8bd2879e
...
...
@@ -20,6 +20,7 @@ limitations under the License. */
#include <utility>
#include <vector>
#include <boost/none.hpp>
#include <boost/variant.hpp>
#include "paddle/fluid/framework/framework.pb.h"
#include "paddle/fluid/framework/op_version_proto.h"
...
...
@@ -30,16 +31,17 @@ namespace framework {
namespace
compatible
{
using
OpAttrVariantT
=
boost
::
variant
<
bool
,
/* AttrType::BOOL */
float
,
/* AttrType::FLOAT */
int32_t
,
/* AttrType::INT */
int64_t
,
/* AttrType::LONG*/
std
::
string
,
/* AttrType::STRING */
std
::
vector
<
bool
>
,
/* AttrType::BOOLS */
std
::
vector
<
float
>
,
/* AttrType::FLOATS */
std
::
vector
<
int32_t
>
,
/* AttrType::INTS */
std
::
vector
<
int64_t
>
,
/* AttrType::LONGS */
std
::
vector
<
std
::
string
>
/* AttrType::STRINGS */
boost
::
variant
<
bool
,
/* AttrType::BOOL */
float
,
/* AttrType::FLOAT */
int32_t
,
/* AttrType::INT */
int64_t
,
/* AttrType::LONG*/
std
::
string
,
/* AttrType::STRING */
std
::
vector
<
bool
>
,
/* AttrType::BOOLS */
std
::
vector
<
float
>
,
/* AttrType::FLOATS */
std
::
vector
<
int32_t
>
,
/* AttrType::INTS */
std
::
vector
<
int64_t
>
,
/* AttrType::LONGS */
std
::
vector
<
std
::
string
>
,
/* AttrType::STRINGS */
boost
::
none_t
/* None */
>
;
struct
OpUpdateInfo
{
...
...
@@ -48,7 +50,7 @@ struct OpUpdateInfo {
struct
OpAttrInfo
:
OpUpdateInfo
{
OpAttrInfo
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
,
const
OpAttrVariantT
&
default_value
)
const
OpAttrVariantT
&
default_value
=
boost
::
none
)
:
name_
{
name
},
default_value_
{
default_value
},
remark_
{
remark
}
{}
const
std
::
string
&
name
()
const
{
return
name_
;
}
...
...
@@ -83,11 +85,18 @@ struct OpBugfixInfo : OpUpdateInfo {
enum
class
OpUpdateType
{
kInvalid
=
0
,
/* Compatibility upgrade */
kModifyAttr
,
kNewAttr
,
kNewInput
,
kNewOutput
,
kBugfixWithBehaviorChanged
,
/* Incompatible upgrade, only for existing registration. */
kDeleteAttr
=
100
,
kModifyInput
,
kModifyOutput
,
kDeleteInput
,
kDeleteOutput
,
};
class
OpUpdateBase
{
...
...
@@ -111,6 +120,7 @@ class OpUpdate : public OpUpdateBase {
class
OpVersionDesc
{
public:
/* Compatibility upgrade */
OpVersionDesc
&&
ModifyAttr
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
,
const
OpAttrVariantT
&
default_value
);
OpVersionDesc
&&
NewAttr
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
,
...
...
@@ -118,10 +128,23 @@ class OpVersionDesc {
OpVersionDesc
&&
NewInput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
NewOutput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
BugfixWithBehaviorChanged
(
const
std
::
string
&
remark
);
/* Incompatible upgrade, only for existing registration. */
OpVersionDesc
&&
DeleteAttr
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
ModifyInput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
ModifyOutput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
DeleteInput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
OpVersionDesc
&&
DeleteOutput
(
const
std
::
string
&
name
,
const
std
::
string
&
remark
);
public:
const
std
::
vector
<
std
::
unique_ptr
<
OpUpdateBase
>>&
infos
()
const
{
return
infos_
;
}
OpVersionDesc
()
=
default
;
OpVersionDesc
(
OpVersionDesc
&&
)
=
default
;
OpVersionDesc
&
operator
=
(
OpVersionDesc
&&
)
=
default
;
...
...
paddle/fluid/framework/op_version_registry_test.cc
浏览文件 @
8bd2879e
...
...
@@ -53,6 +53,19 @@ TEST(test_operator_version, test_operator_version) {
framework
::
compatible
::
OpVersionDesc
()
.
NewInput
(
"X2"
,
"The second input."
)
.
NewOutput
(
"Y2"
,
"The second output."
));
REGISTER_OP_VERSION
(
op_name_0__
)
.
AddCheckpoint
(
R"ROC(
Incompatible upgrade of attribute [height], input [X2] and output [Y2]
)ROC"
,
framework
::
compatible
::
OpVersionDesc
()
.
DeleteAttr
(
"height"
,
"Parameters deleted due to interface alignment."
)
.
ModifyInput
(
"X2"
,
"Modify input due to interface alignment."
)
.
ModifyOutput
(
"Y2"
,
"Modify output due to interface alignment."
)
.
DeleteInput
(
"X2"
,
"Delete input due to interface alignment."
)
.
DeleteOutput
(
"Y2"
,
"Delete output due to interface alignment."
));
}
TEST
(
test_pass_op_version_checker
,
test_pass_op_version_checker
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录