Method TrackDeallocate

tensorflow/core/common_runtime/gpu_memory_planner.cc:201–229

}

void GPUMemoryPlanner::TrackDeallocate(void* ptr) {
  if (!is_stats_.load()) {
    return;
  }
  timeval tmp;
  gettimeofday(&tmp, nullptr);

  AllocStats* alloc_stats;
  {
    std::lock_guard<spin_lock> l(stats_lock_);
    auto iter = ptr_stats_.find(ptr);
    if (iter == ptr_stats_.end()) {
      return;
    }
    alloc_stats = iter->second;
    ptr_stats_.erase(iter);
    alloc_stats_.emplace_back(alloc_stats);
  }
  alloc_stats->end = Timeval2Double(tmp);

  if (SmallAlloc(alloc_stats->size)) {
    GetSmallBin(alloc_stats->size)->TrackDeallocate(alloc_stats);
    return;
  }

  for (auto lifetime_policy : lifetime_stats_polices_) {
    lifetime_policy->TrackDeallocate(alloc_stats);
  }
}

GPULifetimePolicy::GPULifetimePolicy(size_t interval,
    size_t interval_offset, size_t start) :