Searched refs:CUDAStreamGuard (Results 1 – 7 of 7) sorted by relevance
142 struct CUDAStreamGuard { struct144 explicit CUDAStreamGuard() = delete;149 explicit CUDAStreamGuard(Stream stream) : guard_(stream) {} in CUDAStreamGuard() function152 CUDAStreamGuard(const CUDAStreamGuard&) = delete;153 CUDAStreamGuard& operator=(const CUDAStreamGuard&) = delete; argument157 CUDAStreamGuard(CUDAStreamGuard&& other) = delete;158 CUDAStreamGuard& operator=(CUDAStreamGuard&& other) = delete; argument
36 at::cuda::CUDAStreamGuard* guard = in aoti_torch_create_cuda_stream_guard()37 new at::cuda::CUDAStreamGuard(at::cuda::getStreamFromExternal( in aoti_torch_create_cuda_stream_guard()46 { delete reinterpret_cast<at::cuda::CUDAStreamGuard*>(guard); }); in aoti_torch_delete_cuda_stream_guard()
47 called ``setCurrentCUDAStream`` or used ``StreamGuard`` or ``CUDAStreamGuard``.65 …We recommend using ``CUDAStreamGuard``, instead, since it switches to the stream's device and make…66 ``CUDAStreamGuard`` will also restore the current device and stream when it's destroyed68 2. Use ``CUDAStreamGuard`` to switch to a CUDA stream within a scope, it is defined in `CUDAStreamG…111 at::cuda::CUDAStreamGuard guard(myStream);166 at::cuda::CUDAStreamGuard stream_guard(myStream1_1);259 at::cuda::CUDAStreamGuard stream_guard(streams1[1]);
150 at::cuda::CUDAStreamGuard guard(streams1[1]); in TEST()369 at::cuda::CUDAStreamGuard outerGuard(myFirstStream); in TEST()374 at::cuda::CUDAStreamGuard innerGuard(mySecondStream); in TEST()
107 at::cuda::CUDAStreamGuard guard(stream); in allocateTensorForReceiving()
3391 at::cuda::CUDAStreamGuard guard(ncclStream); in checkForNCCLErrorsInternal()3798 at::cuda::CUDAStreamGuard guard(ncclStream); in checkForNCCLErrorsInternal()3932 at::cuda::CUDAStreamGuard guard(ncclStream); in checkForNCCLErrorsInternal()
823 at::cuda::CUDAStreamGuard guard(*op_stream); in collective_post()