Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
477b0c46
P
Paddle
项目概览
Crayon鑫
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
477b0c46
编写于
1月 04, 2021
作者:
S
Shang Zhizhou
提交者:
GitHub
1月 04, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix op version checker of pass bug (#30028) (#30084)
上级
84c2315a
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
14 addition
and
4 deletion
+14
-4
paddle/fluid/framework/ir/quant_conv2d_dequant_fuse_pass.cc
paddle/fluid/framework/ir/quant_conv2d_dequant_fuse_pass.cc
+2
-3
paddle/fluid/framework/op_version_registry.h
paddle/fluid/framework/op_version_registry.h
+12
-1
未找到文件。
paddle/fluid/framework/ir/quant_conv2d_dequant_fuse_pass.cc
浏览文件 @
477b0c46
...
@@ -327,9 +327,8 @@ void QuantDequantFusePass::ApplyImpl(ir::Graph* graph) const {
...
@@ -327,9 +327,8 @@ void QuantDequantFusePass::ApplyImpl(ir::Graph* graph) const {
REGISTER_PASS
(
quant_conv2d_dequant_fuse_pass
,
REGISTER_PASS
(
quant_conv2d_dequant_fuse_pass
,
paddle
::
framework
::
ir
::
QuantDequantFusePass
);
paddle
::
framework
::
ir
::
QuantDequantFusePass
);
REGISTER_PASS_CAPABILITY
(
quant_conv2d_dequant_fuse_pass
);
REGISTER_PASS_CAPABILITY
(
tensorrt_subgraph
_pass
)
REGISTER_PASS_CAPABILITY
(
quant_conv2d_dequant_fuse
_pass
)
.
AddCombination
(
.
AddCombination
(
paddle
::
framework
::
compatible
::
OpVersionComparatorCombination
()
paddle
::
framework
::
compatible
::
OpVersionComparatorCombination
()
.
LE
(
"conv2d"
,
1
)
.
LE
(
"conv2d"
,
1
)
...
@@ -338,5 +337,5 @@ REGISTER_PASS_CAPABILITY(tensorrt_subgraph_pass)
...
@@ -338,5 +337,5 @@ REGISTER_PASS_CAPABILITY(tensorrt_subgraph_pass)
.
EQ
(
"fake_quantize_abs_max"
,
0
)
.
EQ
(
"fake_quantize_abs_max"
,
0
)
.
EQ
(
"fake_quantize_range_abs_max"
,
0
)
.
EQ
(
"fake_quantize_range_abs_max"
,
0
)
.
EQ
(
"fake_quantize_moving_average_abs_max"
,
0
)
.
EQ
(
"fake_quantize_moving_average_abs_max"
,
0
)
.
EQ
(
"fake_channel_wise_quantize_abs_max"
,
0
)
.
LE
(
"fake_channel_wise_quantize_abs_max"
,
1
)
.
EQ
(
"fake_dequantize_max_abs"
,
0
));
.
EQ
(
"fake_dequantize_max_abs"
,
0
));
paddle/fluid/framework/op_version_registry.h
浏览文件 @
477b0c46
...
@@ -240,7 +240,13 @@ class OpVersionComparator {
...
@@ -240,7 +240,13 @@ class OpVersionComparator {
if (OpVersionRegistrar::GetInstance().Has(op_name_)) { \
if (OpVersionRegistrar::GetInstance().Has(op_name_)) { \
version_id = OpVersionRegistrar::GetInstance().version_id(op_name_); \
version_id = OpVersionRegistrar::GetInstance().version_id(op_name_); \
} \
} \
return version_id cmp_math target_version_; \
bool check_ok = version_id cmp_math target_version_; \
if (!check_ok) { \
LOG(WARNING) << "Check op version in pass failed. op name:" \
<< op_name_.c_str() << " op_version:" << version_id \
<< " target_version:" << target_version_; \
} \
return check_ok; \
} \
} \
virtual ~OpVersion##cmp_name##Comparator() {} \
virtual ~OpVersion##cmp_name##Comparator() {} \
\
\
...
@@ -326,6 +332,11 @@ class PassVersionCheckerRegistrar {
...
@@ -326,6 +332,11 @@ class PassVersionCheckerRegistrar {
return
instance
;
return
instance
;
}
}
PassVersionCheckers
&
Register
(
const
std
::
string
&
pass_name
)
{
PassVersionCheckers
&
Register
(
const
std
::
string
&
pass_name
)
{
PADDLE_ENFORCE_EQ
(
pass_version_checkers_map_
.
find
(
pass_name
),
pass_version_checkers_map_
.
end
(),
platform
::
errors
::
AlreadyExists
(
"PassVersionCheckers(%s) has alredy been registered."
,
pass_name
.
c_str
()));
return
pass_version_checkers_map_
[
pass_name
];
return
pass_version_checkers_map_
[
pass_name
];
}
}
bool
IsPassCompatible
(
const
std
::
string
&
fuse_pass_name
)
const
{
bool
IsPassCompatible
(
const
std
::
string
&
fuse_pass_name
)
const
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录