Searched defs:OptionalCUDAStreamGuard (Results 1 – 1 of 1) sorted by relevance
206 struct OptionalCUDAStreamGuard { struct208 explicit OptionalCUDAStreamGuard() : guard_() {} in OptionalCUDAStreamGuard() function213 explicit OptionalCUDAStreamGuard(Stream stream) : guard_(stream) {} in OptionalCUDAStreamGuard() argument218 explicit OptionalCUDAStreamGuard(std::optional<Stream> stream_opt) in OptionalCUDAStreamGuard() function223 OptionalCUDAStreamGuard& operator=(const OptionalCUDAStreamGuard&) = delete; argument229 OptionalCUDAStreamGuard& operator=(OptionalCUDAStreamGuard&& other) = delete; argument236 void reset_stream(Stream stream) { in reset_stream()242 std::optional<CUDAStream> original_stream() const { in original_stream()254 std::optional<CUDAStream> current_stream() const { in current_stream()265 void reset() { in reset()