Lines Matching refs:CUDAStream

60 class C10_CUDA_API CUDAStream {
66   explicit CUDAStream(Stream stream) : stream_(stream) {  in CUDAStream()  function
73   explicit CUDAStream(Unchecked, Stream stream) : stream_(stream) {}  in CUDAStream()  function
75   bool operator==(const CUDAStream& other) const noexcept {
79   bool operator!=(const CUDAStream& other) const noexcept {
165   static CUDAStream unpack3(  in unpack3()
169     return CUDAStream(Stream::unpack3(stream_id, device_index, device_type));  in unpack3()
212 C10_API CUDAStream
215 C10_API CUDAStream
225 C10_API CUDAStream
234 C10_API CUDAStream getDefaultCUDAStream(DeviceIndex device_index = -1);
243 C10_API CUDAStream getCurrentCUDAStream(DeviceIndex device_index = -1);
255 C10_API void setCurrentCUDAStream(CUDAStream stream);
257 C10_API std::ostream& operator<<(std::ostream& stream, const CUDAStream& s);
263 struct hash<c10::cuda::CUDAStream> {
264   size_t operator()(c10::cuda::CUDAStream s) const noexcept {