CUDAStream.h File Reference

#include <cstdint>
#include <utility>
#include <cuda_runtime_api.h>
#include <c10/core/DeviceGuard.h>
#include <c10/core/Stream.h>
#include <c10/cuda/CUDAFunctions.h>
#include <c10/util/Exception.h>

Go to the source code of this file.

Classes
class	c10::cuda::CUDAStream

Namespaces
namespace	c10
namespace	c10::cuda
namespace	std

Functions
CUDAStream	c10::cuda::getStreamFromPool (const bool isHighPriority=false, DeviceIndex device=-1)
	Get a new stream from the CUDA stream pool. More...
CUDAStream	c10::cuda::getStreamFromExternal (cudaStream_t ext_stream, DeviceIndex device_index)
	Get a CUDAStream from a externally allocated one. More...
CUDAStream	c10::cuda::getDefaultCUDAStream (DeviceIndex device_index=-1)
	Get the default CUDA stream, for the passed CUDA device, or for the current device if no device index is passed. More...
CUDAStream	c10::cuda::getCurrentCUDAStream (DeviceIndex device_index=-1)
	Get the current CUDA stream, for the passed CUDA device, or for the current device if no device index is passed. More...
void	c10::cuda::setCurrentCUDAStream (CUDAStream stream)
	Set the current stream on the device of the passed in stream to be the passed in stream. More...
std::ostream &	c10::cuda::operator<< (std::ostream &stream, const CUDAStream &s)