Lines Matching refs:CUDAStream

178   CUDAStream original_stream() const {  in original_stream()
179     return CUDAStream(CUDAStream::UNCHECKED, guard_.original_stream());  in original_stream()
184   CUDAStream current_stream() const {  in current_stream()
185     return CUDAStream(CUDAStream::UNCHECKED, guard_.current_stream());  in current_stream()
242   std::optional<CUDAStream> original_stream() const {  in original_stream()
245       return std::make_optional(CUDAStream(CUDAStream::UNCHECKED, r.value()));  in original_stream()
254   std::optional<CUDAStream> current_stream() const {  in current_stream()
257       return std::make_optional(CUDAStream(CUDAStream::UNCHECKED, r.value()));  in current_stream()
275   explicit CUDAMultiStreamGuard(ArrayRef<CUDAStream> streams)  in CUDAMultiStreamGuard()
291   static std::vector<Stream> unwrapStreams(ArrayRef<CUDAStream> cudaStreams) {  in unwrapStreams()
294     for (const CUDAStream& cudaStream : cudaStreams) {  in unwrapStreams()