Method SyncDeviceContext

paddle/phi/api/lib/context_pool.cc:28–38

namespace paddle::experimental {

void DeviceContextPool::SyncDeviceContext(const Place& place) {
  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;
  }
}

DeviceContextPool& DeviceContextPool::Instance() {
  static DeviceContextPool g_device_context_pool;