Method GetFromPool

include/dali/core/cuda_shared_event.h:56–65

  }

  static CUDASharedEvent GetFromPool(CUDAEventPool &pool, int device_id = -1) {
    if (device_id < 0)
      CUDA_CALL(cudaGetDevice(&device_id));
    CUDAEvent &&event = pool.Get(device_id);
    return CUDASharedEvent(
        std::move(event),
        [device_id, owner = &pool](void *e) {
          owner->Put(CUDAEvent(static_cast<cudaEvent_t>(e)), device_id);
        });
  }

  static CUDASharedEvent GetFromPool(int device_id = -1) {
    return GetFromPool(CUDAEventPool::instance(), device_id);