diff --git a/paddle/fluid/operators/distributed/parameter_prefetch.cc b/paddle/fluid/operators/distributed/parameter_prefetch.cc index 84828c7cd0c3b07ed0ac70c06f4b1821d75e301e..88e87fc63854a81de5caaaff7cee60f63b417f73 100644 --- a/paddle/fluid/operators/distributed/parameter_prefetch.cc +++ b/paddle/fluid/operators/distributed/parameter_prefetch.cc @@ -195,8 +195,8 @@ void prefetchs(const std::vector &id_var_names, const std::vector &endpoints, const framework::ExecutionContext &context, const framework::Scope &scope) { - platform::RecordEvent record_event("Distributed_lookup_table::prefetchs", - platform::EventRole::kInnerOp); + platform::PushEvent("Distributed_lookup_table::prefetchs", + platform::EventRole::kInnerOp); auto vec_dim_1 = 0; auto vec_dim_0 = 0; framework::Variable *var = scope.FindVar(persistable_var_name); @@ -304,6 +304,8 @@ void prefetchs(const std::vector &id_var_names, #endif } } + platform::PopEvent("Distributed_lookup_table::prefetchs", + platform::EventRole::kInnerOp); } }; // namespace distributed diff --git a/paddle/fluid/operators/distributed/parameter_recv.cc b/paddle/fluid/operators/distributed/parameter_recv.cc index ce256cb8b23702b233565d4d22887f076e858f49..65c4b26b0d8b41c4e4292fd1a8dd5064408b98a3 100644 --- a/paddle/fluid/operators/distributed/parameter_recv.cc +++ b/paddle/fluid/operators/distributed/parameter_recv.cc @@ -43,8 +43,8 @@ using DDim = framework::DDim; template void RecvSelectedRows(const CommContext &rpc_ctx, const framework::Scope &scope) { - platform::RecordEvent record_event("ParameterRecv::RecvSelectedRows", - platform::EventRole::kInnerOp); + platform::RecordEvent record_event_grpc("ParameterRecv::RecvSelectedRows", + platform::EventRole::kInnerOp); platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance(); auto cpu_place = platform::CPUPlace(); auto &cpu_ctx = *pool.Get(cpu_place); @@ -114,8 +114,8 @@ void RecvSelectedRows(const CommContext &rpc_ctx, template void RecvLodTensor(const CommContext &rpc_ctx, const framework::Scope &scope) { - platform::RecordEvent record_event("ParameterRecv::RecvLodTensor", - platform::EventRole::kInnerOp); + platform::RecordEvent record_event_grpc("ParameterRecv::RecvLodTensor", + platform::EventRole::kInnerOp); platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance(); auto cpu_place = platform::CPUPlace(); auto &cpu_ctx = *pool.Get(cpu_place); diff --git a/paddle/fluid/operators/distributed/parameter_send.cc b/paddle/fluid/operators/distributed/parameter_send.cc index e95420d4279dcd142b30051360ee3fe4fd41a716..de6ca7d2e350b3bf2a4f85c5e39ee1c320989654 100644 --- a/paddle/fluid/operators/distributed/parameter_send.cc +++ b/paddle/fluid/operators/distributed/parameter_send.cc @@ -98,8 +98,6 @@ template void ParameterSend::operator()(const CommContext &rpc_ctx, const framework::Scope &scope, bool sync, int multi_parts) { - platform::RecordEvent record_event("ParameterSend::operator", - platform::EventRole::kInnerOp); if (rpc_ctx.var_name == STEP_COUNTER) { SendByNotifyRPC(rpc_ctx, scope); return; @@ -117,8 +115,8 @@ void ParameterSend::operator()(const CommContext &rpc_ctx, auto *send_var = scope.FindVar(rpc_ctx.var_name); if (send_var->IsType()) { - platform::RecordEvent record_event("ParameterSend::LoDTensor", - platform::EventRole::kInnerOp); + platform::RecordEvent record_event_grpc("ParameterSend::LoDTensor", + platform::EventRole::kInnerOp); size_t out_num = rpc_ctx.splited_varnames.size(); if (out_num > 1) { auto &send_tensor = send_var->Get(); @@ -167,8 +165,8 @@ void ParameterSend::operator()(const CommContext &rpc_ctx, } } } else if (send_var->IsType()) { - platform::RecordEvent record_event("ParameterSend::SelectedRows", - platform::EventRole::kInnerOp); + platform::RecordEvent record_event_grpc("ParameterSend::SelectedRows", + platform::EventRole::kInnerOp); auto &send_slr = send_var->Get(); auto &send_rows = send_slr.rows(); diff --git a/paddle/fluid/platform/device_tracer.cc b/paddle/fluid/platform/device_tracer.cc index ec934c3b980c3d53ae5581a1317a2610218f2ebf..7cbd05189a1c3469ae6dc51a84d1dffe9a2979c8 100644 --- a/paddle/fluid/platform/device_tracer.cc +++ b/paddle/fluid/platform/device_tracer.cc @@ -694,8 +694,16 @@ Event *CurAnnotation() { return annotation_stack.back(); } std::string CurAnnotationName() { - if (annotation_stack.empty()) return "Unknown"; - return annotation_stack.back()->name(); + std::string curr_annotation_name = "Unknown"; + try { + if (annotation_stack.empty()) { + return curr_annotation_name; + } + return annotation_stack.back()->name(); + } catch (...) { + VLOG(1) << "CurAnnotationName Failed"; + } + return curr_annotation_name; } void SetCurBlock(int block_id) { block_id_stack.push_back(block_id); }