Method Get

paddle/phi/api/lib/context_pool.cc:45–60

}

const phi::DeviceContext* DeviceContextPool::Get(const Place& place) {
  auto it = context_map_.find(place);
  if (it == context_map_.end()) {
    if (!phi::DeviceContextPool::IsInitialized()) {
      phi::memory_utils::InitDevices();
    }
    // only when we need the specific DeviceContext, get and cache it
    auto* dev_ctx = phi::DeviceContextPool::Instance().Get(place);
    {
      std::lock_guard<std::mutex> lock(mutex_);
      context_map_[place] = dev_ctx;
    }
    return dev_ctx;
  }
  return it->second;
}

phi::DeviceContext* DeviceContextPool::GetMutable(const Place& place) {
  return const_cast<phi::DeviceContext*>(Get(place));  // NOLINT