Home
last modified time | relevance | path

Searched defs:streamGuard (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp2792 c10::cuda::CUDAMultiStreamGuard streamGuard(ncclStream); in checkForNCCLErrorsInternal() local
2971 c10::cuda::CUDAMultiStreamGuard streamGuard(ncclStream); in checkForNCCLErrorsInternal() local
3231 c10::cuda::CUDAMultiStreamGuard streamGuard(ncclStream); in checkForNCCLErrorsInternal() local
H A DProcessGroupUCC.cpp831 c10::cuda::CUDAMultiStreamGuard streamGuard(*op_stream); in collective_post() local