Searched defs:CUDAStream (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupNCCL.cpp | 3311 [](at::cuda::CUDAStream&, in checkForNCCLErrorsInternal()
3331 [](at::cuda::CUDAStream&, in checkForNCCLErrorsInternal()
3333 [](at::cuda::CUDAStream&) {}, in checkForNCCLErrorsInternal()
3943 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) {}, in checkForNCCLErrorsInternal()
4315 [](at::cuda::CUDAStream&, in checkForNCCLErrorsInternal()
4316 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) {}, in checkForNCCLErrorsInternal()
4513 [](at::cuda::CUDAStream&, in checkForNCCLErrorsInternal()
4514 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) {}, in checkForNCCLErrorsInternal()
4609 [](at::cuda::CUDAStream&, in checkForNCCLErrorsInternal()
4610 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) {}, in checkForNCCLErrorsInternal()
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAStream.h | 66 explicit CUDAStream(Stream stream) : stream_(stream) { in CUDAStream() function
73 explicit CUDAStream(Unchecked, Stream stream) : stream_(stream) {} in CUDAStream() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/cuda/ |
| H A D | cuda.h | 24 CUDAStream(c10::cuda::CUDAStream s) { in CUDAStream() function
115 friend class CUDAStream; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_stream.h | 27 using CUDAStream = gpu::GpuStream; variable
|
| H A D | cuda_blas.cc | 266 cudaStream_t CUDABlas::CUDAStream(Stream *stream) { in CUDAStream() function in stream_executor::cuda::CUDABlas
|