Method TrackAllocate

tensorflow/core/common_runtime/gpu_memory_planner.cc:175–199

}

void GPUMemoryPlanner::TrackAllocate(size_t alignment, size_t num_bytes, void* ptr) {
  if (!is_stats_.load()) {
    return;
  }

  timeval tmp;
  gettimeofday(&tmp, nullptr);

  auto alloc_stats = new AllocStats;
  alloc_stats->begin = Timeval2Double(tmp);
  alloc_stats->size = num_bytes;
  {
    std::lock_guard<spin_lock> l(stats_lock_);
    ptr_stats_[ptr] = alloc_stats;
  }

  if (SmallAlloc(num_bytes)) {
    GetSmallBin(num_bytes)->TrackAllocate(alignment);
    return;
  }

  for (auto lifetime_policy : lifetime_stats_polices_) {
    lifetime_policy->TrackAllocate(alignment, num_bytes);
  }
}

void GPUMemoryPlanner::TrackDeallocate(void* ptr) {
  if (!is_stats_.load()) {