From c0ddb93ccc84be81cbbffd1f18c6b19ba154dc86 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Fri, 15 Mar 2019 21:06:16 +0800 Subject: [PATCH] Polish code test=develop --- paddle/fluid/framework/grad_op_desc_maker.h | 7 ++----- paddle/fluid/imperative/profiler.cc | 3 --- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/paddle/fluid/framework/grad_op_desc_maker.h b/paddle/fluid/framework/grad_op_desc_maker.h index 46ebf4051fb..3645cd4e432 100644 --- a/paddle/fluid/framework/grad_op_desc_maker.h +++ b/paddle/fluid/framework/grad_op_desc_maker.h @@ -55,14 +55,11 @@ class GradOpDescMakerBase { std::back_inserter(ret_val), [this](const std::string& fwd_var_name) -> std::string { auto g_name = GradVarName(fwd_var_name); - if (no_grad_set_.empty()) { + if (no_grad_set_.empty() || !no_grad_set_.count(g_name)) { (*this->grad_to_var_)[g_name] = fwd_var_name; return g_name; - } else if (no_grad_set_.count(g_name)) { - return kEmptyVarName; } else { - (*this->grad_to_var_)[g_name] = fwd_var_name; - return g_name; + return kEmptyVarName; } }); if (!drop_empty_grad) { diff --git a/paddle/fluid/imperative/profiler.cc b/paddle/fluid/imperative/profiler.cc index 828c36c5aef..34570b3a60e 100644 --- a/paddle/fluid/imperative/profiler.cc +++ b/paddle/fluid/imperative/profiler.cc @@ -36,13 +36,11 @@ static bool gTracerProfilerStarted = false; #endif void StartProfile() { - LOG(ERROR) << "XX " << FLAGS_tracer_profile_fname; if (!FLAGS_tracer_profile_fname.empty()) { std::call_once(gTracerProfileOnce, [] { #ifdef WITH_GPERFTOOLS ProfilerStart(FLAGS_tracer_profile_fname.c_str()); gTracerProfilerStarted = true; - LOG(ERROR) << "YY"; #else LOG(WARNING) << "Paddle is not compiled with gperftools. " "FLAGS_tracer_profile_fname will be ignored"; @@ -52,7 +50,6 @@ void StartProfile() { } void StopProfile() { - LOG(ERROR) << "ZZ " << FLAGS_tracer_profile_fname; #ifdef WITH_GPERFTOOLS ProfilerFlush(); #else -- GitLab