Searched refs:GpuStream (Results 1 – 10 of 10) sorted by relevance
25 bool GpuStream::Init() { in Init()35 void GpuStream::Destroy() { in Destroy()47 bool GpuStream::IsIdle() const { in IsIdle()51 GpuStream* AsGpuStream(Stream* stream) { in AsGpuStream()53 return static_cast<GpuStream*>(stream->implementation()); in AsGpuStream()
35 class GpuStream : public internal::StreamInterface {37 explicit GpuStream(GpuExecutor* parent) in GpuStream() function41 ~GpuStream() override {} in ~GpuStream()91 GpuStream* AsGpuStream(Stream* stream);
30 class GpuStream; variable54 bool Start(GpuStream* stream);57 bool Stop(GpuStream* stream);
74 bool GpuTimer::Start(GpuStream* stream) { in Start()83 bool GpuTimer::Stop(GpuStream* stream) { in Stop()
43 port::Status Record(GpuStream* stream);
39 port::Status GpuEvent::Record(GpuStream* stream) { in Record()
27 using CUDAStream = gpu::GpuStream;
704 GpuStream* cuda_stream = AsGpuStream(stream); in DeallocateStream()882 return std::unique_ptr<internal::StreamInterface>(new GpuStream(this)); in GetStreamImplementation()
612 GpuStream* rocm_stream = AsGpuStream(stream); in DeallocateStream()787 return std::unique_ptr<internal::StreamInterface>(new GpuStream(this)); in GetStreamImplementation()
364 static_cast<stream_executor::gpu::GpuStream*>(stream->implementation()) in GetStream()