Method DmaMemoryPool

core/memory.cc:269–304

}

DmaMemoryPool::DmaMemoryPool(size_t size, int socket_id)
    : initialized_(false), socket_id_(socket_id), total_free_bytes_(0) {
  CHECK_GT(size, 0);
  CHECK_GE(socket_id, -1);

  // Try 1GB hugepages first, then 2MB ones.
  HugepageSize page_size = HugepageSize::k1GB;

  while (total_free_bytes_ < size) {
    size_t page_bytes = static_cast<size_t>(page_size);
    void *ptr = AllocHugepageFromSocket(page_size, socket_id);

    if (ptr != nullptr) {
      total_free_bytes_ += page_bytes;
      AddRegion(reinterpret_cast<uintptr_t>(ptr), page_bytes);
      pages_.push_back(ptr);
    } else {
      if (page_size == HugepageSize::k1GB) {
        page_size = HugepageSize::k2MB;
      } else {
        break;
      }
    }
  }

  if (total_free_bytes_ >= size) {
    initialized_ = true;
    return;
  }

  // Failed. Give up and clean up.
  for (auto *ptr : pages_) {
    FreeHugepage(ptr);
  }
  pages_.clear();
}

DmaMemoryPool::~DmaMemoryPool() {
  for (auto *ptr : pages_) {