Fix nullptr when doing nested profileing

panyx0718 · Mar 11, 2018 · a4b801c · a4b801c
1 parent 6272b57
commit a4b801c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/paddle/fluid/platform/device_tracer.cc b/paddle/fluid/platform/device_tracer.cc
@@ -192,6 +192,12 @@ class DeviceTracerImpl : public DeviceTracer {
   }
 
   void AddCPURecords(const char *anno, uint64_t start_ns, uint64_t end_ns) {
+    if (!anno) {
+      // TODO(panyx0718): Currently, it doesn't support nested situation
+      // Up-level can be cleared by low-level and therefore get nullptr
+      // here.
+      return;
+    }
     std::lock_guard<std::mutex> l(trace_mu_);
     cpu_records_.push_back(
         CPURecord{anno, start_ns, end_ns,