Searched defs:CUDAStreamGuard (Results 1 – 1 of 1) sorted by relevance
142 struct CUDAStreamGuard { struct149 explicit CUDAStreamGuard(Stream stream) : guard_(stream) {} in CUDAStreamGuard() argument153 CUDAStreamGuard& operator=(const CUDAStreamGuard&) = delete; argument158 CUDAStreamGuard& operator=(CUDAStreamGuard&& other) = delete; argument172 void reset_stream(Stream stream) { in reset_stream()178 CUDAStream original_stream() const { in original_stream()206 struct OptionalCUDAStreamGuard { argument