未验证 提交 5b1ad140 编写于 作者: D danleifeng 提交者: GitHub

edit pten_util vlog print level;test=develop (#37828)

上级 364376e5
...@@ -136,17 +136,17 @@ KernelArgsNameMakerByOpProto::GetInputArgsNames() { ...@@ -136,17 +136,17 @@ KernelArgsNameMakerByOpProto::GetInputArgsNames() {
auto& in = op_proto_->inputs()[i]; auto& in = op_proto_->inputs()[i];
auto& in_name = in.name(); auto& in_name = in.name();
if ((in.has_extra() && in.extra()) || (in.has_quant() && in.quant())) { if ((in.has_extra() && in.extra()) || (in.has_quant() && in.quant())) {
VLOG(1) << "Parse PtenKernel input: skip extra & quant input - " VLOG(3) << "Parse PtenKernel input: skip extra & quant input - "
<< in_name; << in_name;
continue; continue;
} }
// If contains dispensable input, we should override the // If contains dispensable input, we should override the
// GetExpectedPtenKernelArgs method self // GetExpectedPtenKernelArgs method self
if (in.has_dispensable() && in.dispensable()) { if (in.has_dispensable() && in.dispensable()) {
VLOG(1) << "Parse PtenKernel input: skip dispensable input - " << in_name; VLOG(3) << "Parse PtenKernel input: skip dispensable input - " << in_name;
continue; continue;
} }
VLOG(1) << "Parse PtenKernel input: " << in_name; VLOG(3) << "Parse PtenKernel input: " << in_name;
input_names_.emplace_back(in_name); input_names_.emplace_back(in_name);
} }
return input_names_; return input_names_;
...@@ -158,7 +158,7 @@ KernelArgsNameMakerByOpProto::GetOutputArgsNames() { ...@@ -158,7 +158,7 @@ KernelArgsNameMakerByOpProto::GetOutputArgsNames() {
auto& out = op_proto_->outputs()[i]; auto& out = op_proto_->outputs()[i];
auto& out_name = out.name(); auto& out_name = out.name();
// TODO(chenweihang): outputs also need skip some cases // TODO(chenweihang): outputs also need skip some cases
VLOG(1) << "Parse PtenKernel output: " << out_name; VLOG(3) << "Parse PtenKernel output: " << out_name;
output_names_.emplace_back(out_name); output_names_.emplace_back(out_name);
} }
return output_names_; return output_names_;
...@@ -172,17 +172,17 @@ KernelArgsNameMakerByOpProto::GetAttrsArgsNames() { ...@@ -172,17 +172,17 @@ KernelArgsNameMakerByOpProto::GetAttrsArgsNames() {
if (attr_name == "use_mkldnn" || attr_name == "op_role" || if (attr_name == "use_mkldnn" || attr_name == "op_role" ||
attr_name == "op_role_var" || attr_name == "op_namescope" || attr_name == "op_role_var" || attr_name == "op_namescope" ||
attr_name == "op_callstack" || attr_name == "op_device") { attr_name == "op_callstack" || attr_name == "op_device") {
VLOG(1) << "Parse PtenKernel attribute: skip needless attr - " VLOG(3) << "Parse PtenKernel attribute: skip needless attr - "
<< attr_name; << attr_name;
continue; continue;
} }
if ((attr.has_extra() && attr.extra()) || if ((attr.has_extra() && attr.extra()) ||
(attr.has_quant() && attr.quant())) { (attr.has_quant() && attr.quant())) {
VLOG(1) << "Parse PtenKernel attribute: skip extra & quant attr - " VLOG(3) << "Parse PtenKernel attribute: skip extra & quant attr - "
<< attr_name; << attr_name;
continue; continue;
} }
VLOG(1) << "Parse PtenKernel attribute: " << attr_name; VLOG(3) << "Parse PtenKernel attribute: " << attr_name;
attr_names_.emplace_back(attr_name); attr_names_.emplace_back(attr_name);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册