From 8fba68d3a6baca1116ecba8b90bb15a22064d15a Mon Sep 17 00:00:00 2001
From: chenjian <chenjian26@baidu.com>
Date: Thu, 7 Apr 2022 11:01:20 +0800
Subject: [PATCH] Fix dygraph record event position (#41445)

* no

* maintain old profiler

* fix old dygraph record event
---
 paddle/fluid/imperative/partial_grad_engine.cc         | 3 ++-
 paddle/fluid/imperative/tracer.cc                      | 2 +-
 paddle/fluid/platform/profiler/chrometracing_logger.cc | 6 +++---
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/paddle/fluid/imperative/partial_grad_engine.cc b/paddle/fluid/imperative/partial_grad_engine.cc
index 56ddbf3386..f2f64d92a2 100644
--- a/paddle/fluid/imperative/partial_grad_engine.cc
+++ b/paddle/fluid/imperative/partial_grad_engine.cc
@@ -826,6 +826,8 @@ std::vector<std::shared_ptr<VarBase>> PartialGradTask::Run() {
 }
 
 void PartialGradTask::RunEachOp(OpBase *op) {
+  platform::RecordEvent op_type_record_event(
+      op->Type() + " grad trace_op", platform::TracerEventType::Operator, 1);
   // Prepare new inputs
   NameVarMap<VarBase> tmp_ins;
   for (auto &input_pair : op->GetInsMap()) {
@@ -908,7 +910,6 @@ void PartialGradTask::RunEachOp(OpBase *op) {
   // Run op
   OpBase::Run(op->InnerOp(), tmp_ins, tmp_outs, op->Attrs(),
               op->DefaultAttrsMap(), op->place());
-
   if (create_graph_) {
     auto double_grad_node =
         CreateGradOpNode(op->InnerOp(), tmp_ins, tmp_outs, op->Attrs(),
diff --git a/paddle/fluid/imperative/tracer.cc b/paddle/fluid/imperative/tracer.cc
index 3d4cfa2df3..6b20b9b393 100644
--- a/paddle/fluid/imperative/tracer.cc
+++ b/paddle/fluid/imperative/tracer.cc
@@ -313,7 +313,7 @@ void Tracer::TraceOpImpl(const std::string& type,
 
   {
     platform::RecordEvent node_creation_record_event(
-        type + " node_creation", platform::TracerEventType::Operator, 1);
+        type + " node_creation", platform::TracerEventType::OperatorInner, 1);
 
     if (ComputeRequiredGrad(new_ins, outs, trace_backward)) {
       PADDLE_ENFORCE_EQ(
diff --git a/paddle/fluid/platform/profiler/chrometracing_logger.cc b/paddle/fluid/platform/profiler/chrometracing_logger.cc
index 0c5d90a927..d7879e7be5 100644
--- a/paddle/fluid/platform/profiler/chrometracing_logger.cc
+++ b/paddle/fluid/platform/profiler/chrometracing_logger.cc
@@ -325,7 +325,7 @@ void ChromeTracingLogger::HandleTypeKernel(
     "name": "%s[%s]", "pid": %lld, "tid": %lld,
     "ts": %lld, "dur": %.3f,
     "ph": "X", "cat": "%s", 
-    "cname": "rail_animation",
+    "cname": "cq_build_failed",
     "args": {
       "start_time": "%.3f us",
       "end_time": "%.3f us",
@@ -376,7 +376,7 @@ void ChromeTracingLogger::HandleTypeMemcpy(
     "name": "%s[%s]", "pid": %lld, "tid": %lld,
     "ts": %lld, "dur": %.3f,
     "ph": "X", "cat": "%s", 
-    "cname": "rail_animation",
+    "cname": "cq_build_failed",
     "args": {
       "start_time": "%.3f us",
       "end_time": "%.3f us",
@@ -411,7 +411,7 @@ void ChromeTracingLogger::HandleTypeMemset(
     "name": "%s[%s]", "pid": %lld, "tid": %lld,
     "ts": %lld, "dur": %.3f,
     "ph": "X", "cat": "%s", 
-    "cname": "rail_animation",
+    "cname": "cq_build_failed",
     "args": {
       "start_time": "%.3f us",
       "end_time": "%.3f us",
-- 
GitLab