未验证 提交 05643dc3 编写于 作者: 王明冬 提交者: GitHub

[pass_enhance]add global extra attributes for op def, test=develop (#34009)

上级 80bd093a
...@@ -19,6 +19,15 @@ limitations under the License. */ ...@@ -19,6 +19,15 @@ limitations under the License. */
#include "paddle/fluid/framework/op_def_api.h" #include "paddle/fluid/framework/op_def_api.h"
#include "paddle/fluid/framework/op_info.h" #include "paddle/fluid/framework/op_info.h"
namespace {
std::unordered_set<std::string> global_extra_attrs = {
"op_role", "op_role_var", "op_namescope",
"op_callstack", "op_device", "@ENABLE_CACHE_RUNTIME_CONTEXT@",
"is_test", "use_mkldnn", "mkldnn_data_type",
"use_quantizer", "mkldnn_data_type", "use_cudnn",
"name"};
}
namespace paddle { namespace paddle {
namespace framework { namespace framework {
namespace ir { namespace ir {
...@@ -171,7 +180,8 @@ bool OpCompat::Judge(const OpDesc& op_desc) { ...@@ -171,7 +180,8 @@ bool OpCompat::Judge(const OpDesc& op_desc) {
for (auto& attr_map : op_desc.GetAttrMap()) { for (auto& attr_map : op_desc.GetAttrMap()) {
if (attr_compats_.find(attr_map.first) == attr_compats_.end()) { if (attr_compats_.find(attr_map.first) == attr_compats_.end()) {
if (extra_attrs_.find(attr_map.first) != extra_attrs_.end()) { if (global_extra_attrs.find(attr_map.first) != global_extra_attrs.end() ||
extra_attrs_.find(attr_map.first) != extra_attrs_.end()) {
continue; continue;
} }
if (!AttrCompat(attr_map.first, this).IsLeftDefault()(op_desc)) { if (!AttrCompat(attr_map.first, this).IsLeftDefault()(op_desc)) {
......
...@@ -17,25 +17,3 @@ def { ...@@ -17,25 +17,3 @@ def {
name: "Out" name: "Out"
} }
} }
extra {
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
...@@ -15,6 +15,9 @@ def { ...@@ -15,6 +15,9 @@ def {
inputs { inputs {
name: "Variance" name: "Variance"
} }
inputs {
name: "MomentumTensor"
}
outputs { outputs {
name: "Y" name: "Y"
} }
...@@ -39,29 +42,18 @@ def { ...@@ -39,29 +42,18 @@ def {
} }
} }
extra { extra {
inputs {
name: "MomentumTensor"
}
attrs {
name: "@ENABLE_CACHE_RUNTIME_CONTEXT@"
type: BOOLEAN
}
attrs { attrs {
name: "is_test" name: "momentum"
type: BOOLEAN type: FLOAT
} }
attrs { attrs {
name: "momentum" name: "Y0_threshold"
type: FLOAT type: FLOAT
} }
attrs { attrs {
name: "data_layout" name: "data_layout"
type: STRING type: STRING
} }
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "fuse_with_relu" name: "fuse_with_relu"
type: BOOLEAN type: BOOLEAN
...@@ -74,25 +66,5 @@ extra { ...@@ -74,25 +66,5 @@ extra {
name: "trainable_statistics" name: "trainable_statistics"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -14,37 +14,3 @@ def { ...@@ -14,37 +14,3 @@ def {
type: INT type: INT
} }
} }
extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
...@@ -45,6 +45,14 @@ extra { ...@@ -45,6 +45,14 @@ extra {
name: "Input_scale" name: "Input_scale"
type: FLOAT type: FLOAT
} }
attrs {
name: "Input0_threshold"
type: FLOAT
}
attrs {
name: "weight_scale"
type: FLOAT
}
attrs { attrs {
name: "quantization_type" name: "quantization_type"
type: STRING type: STRING
...@@ -57,42 +65,14 @@ extra { ...@@ -57,42 +65,14 @@ extra {
name: "out_threshold" name: "out_threshold"
type: FLOAT type: FLOAT
} }
attrs {
name: "@ENABLE_CACHE_RUNTIME_CONTEXT@"
type: BOOLEAN
}
attrs { attrs {
name: "skip_quant" name: "skip_quant"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "name"
type: STRING
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs { attrs {
name: "fuse_relu_before_depthwise_conv" name: "fuse_relu_before_depthwise_conv"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "fuse_relu" name: "fuse_relu"
type: BOOLEAN type: BOOLEAN
...@@ -153,25 +133,5 @@ extra { ...@@ -153,25 +133,5 @@ extra {
name: "exhaustive_search" name: "exhaustive_search"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -46,26 +46,10 @@ def { ...@@ -46,26 +46,10 @@ def {
} }
} }
extra { extra {
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "force_fp32_output" name: "force_fp32_output"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "fuse_relu" name: "fuse_relu"
type: BOOLEAN type: BOOLEAN
...@@ -86,25 +70,5 @@ extra { ...@@ -86,25 +70,5 @@ extra {
name: "workspace_size_MB" name: "workspace_size_MB"
type: INT type: INT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -6,6 +6,9 @@ def { ...@@ -6,6 +6,9 @@ def {
inputs { inputs {
name: "Filter" name: "Filter"
} }
inputs {
name: "ResidualData"
}
outputs { outputs {
name: "Output" name: "Output"
} }
...@@ -35,33 +38,10 @@ def { ...@@ -35,33 +38,10 @@ def {
} }
} }
extra { extra {
inputs {
name: "ResidualData"
}
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs { attrs {
name: "fuse_relu_before_depthwise_conv" name: "fuse_relu_before_depthwise_conv"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "fuse_relu" name: "fuse_relu"
type: BOOLEAN type: BOOLEAN
......
...@@ -14,26 +14,4 @@ def { ...@@ -14,26 +14,4 @@ def {
type: BOOLEAN type: BOOLEAN
} }
} }
extra {
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
...@@ -57,42 +57,14 @@ extra { ...@@ -57,42 +57,14 @@ extra {
name: "out_threshold" name: "out_threshold"
type: FLOAT type: FLOAT
} }
attrs {
name: "@ENABLE_CACHE_RUNTIME_CONTEXT@"
type: BOOLEAN
}
attrs { attrs {
name: "skip_quant" name: "skip_quant"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "name"
type: STRING
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs { attrs {
name: "fuse_relu_before_depthwise_conv" name: "fuse_relu_before_depthwise_conv"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "fuse_relu" name: "fuse_relu"
type: BOOLEAN type: BOOLEAN
...@@ -153,25 +125,5 @@ extra { ...@@ -153,25 +125,5 @@ extra {
name: "exhaustive_search" name: "exhaustive_search"
type: BOOLEAN type: BOOLEAN
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -15,10 +15,6 @@ def { ...@@ -15,10 +15,6 @@ def {
} }
} }
extra { extra {
attrs {
name: "@ENABLE_CACHE_RUNTIME_CONTEXT@"
type: BOOLEAN
}
attrs { attrs {
name: "out_threshold" name: "out_threshold"
type: FLOAT type: FLOAT
...@@ -37,19 +33,6 @@ extra { ...@@ -37,19 +33,6 @@ extra {
type: STRING type: STRING
# no longer to use # no longer to use
} }
attrs {
name: "use_quantizer"
type: BOOLEAN
# no longer to use, Use 'mkldnn_data_type' instead.
}
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "Scale_x" name: "Scale_x"
type: FLOAT type: FLOAT
...@@ -62,24 +45,4 @@ extra { ...@@ -62,24 +45,4 @@ extra {
name: "Scale_out" name: "Scale_out"
type: FLOAT type: FLOAT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -15,10 +15,6 @@ def { ...@@ -15,10 +15,6 @@ def {
} }
} }
extra { extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "x_data_format" name: "x_data_format"
type: STRING type: STRING
...@@ -27,14 +23,6 @@ extra { ...@@ -27,14 +23,6 @@ extra {
name: "y_data_format" name: "y_data_format"
type: STRING type: STRING
} }
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "Scale_x" name: "Scale_x"
type: FLOAT type: FLOAT
...@@ -47,26 +35,6 @@ extra { ...@@ -47,26 +35,6 @@ extra {
name: "Scale_out" name: "Scale_out"
type: FLOAT type: FLOAT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
attrs { attrs {
name: "act" name: "act"
type: STRING type: STRING
......
...@@ -15,10 +15,6 @@ def { ...@@ -15,10 +15,6 @@ def {
} }
} }
extra { extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "x_data_format" name: "x_data_format"
type: STRING type: STRING
...@@ -27,14 +23,6 @@ extra { ...@@ -27,14 +23,6 @@ extra {
name: "y_data_format" name: "y_data_format"
type: STRING type: STRING
} }
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "Scale_x" name: "Scale_x"
type: FLOAT type: FLOAT
...@@ -47,24 +35,4 @@ extra { ...@@ -47,24 +35,4 @@ extra {
name: "Scale_out" name: "Scale_out"
type: FLOAT type: FLOAT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
} }
...@@ -15,10 +15,6 @@ def { ...@@ -15,10 +15,6 @@ def {
} }
} }
extra { extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "x_data_format" name: "x_data_format"
type: STRING type: STRING
...@@ -27,14 +23,6 @@ extra { ...@@ -27,14 +23,6 @@ extra {
name: "y_data_format" name: "y_data_format"
type: STRING type: STRING
} }
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "Scale_x" name: "Scale_x"
type: FLOAT type: FLOAT
...@@ -47,26 +35,6 @@ extra { ...@@ -47,26 +35,6 @@ extra {
name: "Scale_out" name: "Scale_out"
type: FLOAT type: FLOAT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
attrs { attrs {
name: "act" name: "act"
type: STRING type: STRING
......
...@@ -15,10 +15,6 @@ def { ...@@ -15,10 +15,6 @@ def {
} }
} }
extra { extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs { attrs {
name: "x_data_format" name: "x_data_format"
type: STRING type: STRING
...@@ -27,14 +23,6 @@ extra { ...@@ -27,14 +23,6 @@ extra {
name: "y_data_format" name: "y_data_format"
type: STRING type: STRING
} }
attrs {
name: "use_quantizer"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs { attrs {
name: "Scale_x" name: "Scale_x"
type: FLOAT type: FLOAT
...@@ -47,26 +35,6 @@ extra { ...@@ -47,26 +35,6 @@ extra {
name: "Scale_out" name: "Scale_out"
type: FLOAT type: FLOAT
} }
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
attrs { attrs {
name: "act" name: "act"
type: STRING type: STRING
......
...@@ -18,30 +18,3 @@ def { ...@@ -18,30 +18,3 @@ def {
type: INT type: INT
} }
} }
extra {
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
...@@ -18,29 +18,3 @@ def { ...@@ -18,29 +18,3 @@ def {
type: INT type: INT
} }
} }
extra {
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
}
...@@ -8,9 +8,9 @@ def { ...@@ -8,9 +8,9 @@ def {
} }
} }
extra { extra {
attrs { attrs {
name: "@ENABLE_CACHE_RUNTIME_CONTEXT@" name: "X0_threshold"
type: BOOLEAN type: FLOAT
} }
attrs { attrs {
name: "X0_threshold" name: "X0_threshold"
...@@ -24,40 +24,4 @@ extra { ...@@ -24,40 +24,4 @@ extra {
name: "Out0_threshold" name: "Out0_threshold"
type: FLOAT type: FLOAT
} }
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
attrs {
name: "op_role"
type: INT
}
attrs {
name: "op_role_var"
type: STRINGS
}
attrs {
name: "op_namescope"
type: STRING
}
attrs {
name: "op_callstack"
type: STRINGS
}
attrs {
name: "op_device"
type: STRING
}
attrs {
name: "is_test"
type: BOOLEAN
}
attrs {
name: "name"
type: STRINGS
}
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册