Method BigAllocate

tensorflow/core/common_runtime/tensorpool_allocator.cc:328–356

}

void* TensorPoolAllocator::BigAllocate(size_t alignment,
    size_t num_bytes) {
  auto header_size = std::max(sizeof(Header), alignment);
  auto total = num_bytes + header_size; 
  if (!inited_.load()) {
    mem_planner_->TrackAllocate(alignment, total);
    auto ptr = sub_allocator_->Alloc(alignment, total);
    return SetDefaultHeader(!inited_.load(), ptr, total, header_size);
  }

  auto id = Index(total, alignment_, alignment_offset_);
  if (unlikely(id < 0)) {
    auto ptr = sub_allocator_->Alloc(alignment, total);
    return SetDefaultHeader(!inited_.load(), ptr, total, header_size);
  }

  auto b = GetBin(id);
  if (unlikely(b == nullptr)) {
    auto ptr = sub_allocator_->Alloc(alignment, total);
    return SetDefaultHeader(!inited_.load(), ptr, total, header_size);
  }

  auto ptr = b->Allocate(total, header_size);
  if (likely(ptr != nullptr)) {
    return ptr;
  }
  ptr = sub_allocator_->Alloc(alignment, total);
  return SetDefaultHeader(!inited_.load(), ptr, total, header_size);
}

// unlikely execute this path which do some atomic operations
void* TensorPoolAllocator::BigAllocateStatistic(size_t alignment,