Searched refs:default_stream (Results 1 – 8 of 8) sorted by relevance
33 …ITH_EXCEPT_NOTRACE(cudnnSetStream(cudnn_handle_, reinterpret_cast<cudaStream_t>(default_stream())), in InitDevice()37 cublasSetStream(cublas_handle_, reinterpret_cast<cudaStream_t>(default_stream())), in InitDevice()42 cusolverDnSetStream(cusolver_dn_handle_, reinterpret_cast<cudaStream_t>(default_stream())), in InitDevice()78 const CudaDeviceStream &GPUDeviceManager::default_stream() const { return default_stream_; } in default_stream() function in mindspore::device::gpu::GPUDeviceManager
50 auto &stream = GPUDeviceManager::GetInstance().default_stream(); in SyncDeviceToHost()85 auto &stream = GPUDeviceManager::GetInstance().default_stream(); in SyncHostToDevice()113 auto &stream = GPUDeviceManager::GetInstance().default_stream(); in SyncHostToDevice()
43 const CudaDeviceStream &default_stream() const;
48 auto &stream = GPUDeviceManager::GetInstance().default_stream(); in MallocContinuousMemFromMemPool()
39 CudaDeviceStream compute_stream = GPUDeviceManager::GetInstance().default_stream(); in AssignGpuStream()
217 auto &stream = GPUDeviceManager::GetInstance().default_stream(); in MemcpyAsync()249 stream_ = GPUDeviceManager::GetInstance().default_stream(); in InitDevice()
102 explicit stream_holder( std::ostream& default_stream = std::cout ) in stream_holder() argument103 : m_stream( &default_stream ) in stream_holder()
127 auto stream = GPUDeviceManager::GetInstance().default_stream(); in InitDevice()