Method Init

tensorflow/core/common_runtime/tensorpool_allocator.cc:112–149

}

void TensorPoolAllocator::Init() {
  bool tmp = false;
  if (initing_.compare_exchange_strong(tmp, true)) {
    signal(SIGUSR1, signal_handler);
    auto lifetime_policy = mem_planner_->BestLifetimePolicy();
    lifetime_policy->Dump();

    alignment_ = lifetime_policy->Alignment();
    alignment_offset_ = lifetime_policy->AlignmentOffset();

    auto policy_large_bins = lifetime_policy->GetLargeBins();
    for (auto rit = policy_large_bins.rbegin();
        rit != policy_large_bins.rend(); ++rit) {
      auto bin_info = rit->second;
      auto bin = new Bin(bin_info->BlockSize(), bin_info->ChunkSize(),
          bin_info->Alignment(), bin_info->VBlocks(), 
          sub_allocator_.get(), this);
      large_lifetime_bins_.emplace(rit->first, bin);
    }
    auto policy_bins = lifetime_policy->GetBins();
    large_bin_index_ = policy_bins.size();
    lifetime_bins_.resize(large_bin_index_);

    // create bigger bin first
    for (auto it = policy_bins.rbegin(); it != policy_bins.rend();
        ++it) {
      Bin* bin = nullptr;
      if ((*it)->BlockSize() > 0 || (*it)->VBlocks().size() > 0) {
        bin = new Bin((*it)->BlockSize(), (*it)->ChunkSize(),
            (*it)->Alignment(), (*it)->VBlocks(), 
            sub_allocator_.get(), this);
      }
      lifetime_bins_[(*it)->BinIndex()] = bin;
    }
    LOG(INFO) << "TensorPoolAllocator enabled";
    inited_ = true;
  }
}

void* TensorPoolAllocator::AllocateRaw(size_t alignment,
    size_t num_bytes) {