Searched refs:gpu_stream_ (Results 1 – 2 of 2) sorted by relevance
38 : parent_(parent), gpu_stream_(nullptr), completed_event_(nullptr) {} in GpuStream()43 void* GpuStreamHack() override { return gpu_stream_; } in GpuStreamHack()45 return reinterpret_cast<void**>(&gpu_stream_); in GpuStreamMemberHack()69 DCHECK(gpu_stream_ != nullptr); in gpu_stream()70 return const_cast<GpuStreamHandle>(gpu_stream_); in gpu_stream()80 GpuStreamHandle gpu_stream_; // Wrapped CUDA stream handle. variable
26 if (!GpuDriver::CreateStream(parent_->gpu_context(), &gpu_stream_)) { in Init()43 GpuDriver::DestroyStream(parent_->gpu_context(), &gpu_stream_); in Destroy()47 return GpuDriver::IsStreamIdle(parent_->gpu_context(), gpu_stream_); in IsIdle()