Searched refs:gpu_stream_ (Results 1 – 5 of 5) 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()71 DCHECK(gpu_stream_ != nullptr); in gpu_stream()72 return const_cast<GpuStreamHandle>(gpu_stream_); in gpu_stream()82 GpuStreamHandle gpu_stream_; // Wrapped CUDA stream handle. variable
26 if (!GpuDriver::CreateStream(parent_->gpu_context(), &gpu_stream_, in Init()44 GpuDriver::DestroyStream(parent_->gpu_context(), &gpu_stream_); in Destroy()48 return GpuDriver::IsStreamIdle(parent_->gpu_context(), gpu_stream_); in IsIdle()
133 gpu_stream_ = *hip_stream_ptr; in GpuSparse()140 auto it = handle_map->find(gpu_stream_); in Initialize()142 LOG(INFO) << "Creating GpuSparse handles for stream " << gpu_stream_; in Initialize()145 HipSparseHandles new_handles(gpu_stream_); in Initialize()147 it = handle_map->insert(std::make_pair(gpu_stream_, std::move(new_handles))) in Initialize()
161 gpu_stream_ = *cuda_stream_ptr; in GpuSparse()168 auto it = handle_map->find(gpu_stream_); in Initialize()170 LOG(INFO) << "Creating CudaSparse handles for stream " << gpu_stream_; in Initialize()173 CudaSparseHandles new_handles(gpu_stream_); in Initialize()175 it = handle_map->insert(std::make_pair(gpu_stream_, std::move(new_handles))) in Initialize()
453 gpuStream_t gpu_stream_; variable