Public Member Functions
	CUDAMultiStreamGuard (ArrayRef< CUDAStream > streams)
	Calls `set_stream` on each of the streams in the list. More...

	CUDAMultiStreamGuard (const CUDAGuard &)=delete

CUDAMultiStreamGuard &	operator= (const CUDAGuard &)=delete

	CUDAMultiStreamGuard (CUDAGuard &&other)=delete

CUDAMultiStreamGuard &	operator= (CUDAGuard &&other)=delete

ArrayRef< CUDAStream >	original_streams () const

	~CUDAMultiStreamGuard ()
	Resets the CUDA stream on each device to the one that was active upon construction. More...

Detailed Description

Definition at line 13 of file CUDAMultiStreamGuard.h.

Constructor & Destructor Documentation

at::cuda::CUDAMultiStreamGuard::CUDAMultiStreamGuard ( ArrayRef< CUDAStream > streams )

inlineexplicit

Calls set_stream on each of the streams in the list.

This may be useful if you need to set different streams for different devices.

Definition at line 18 of file CUDAMultiStreamGuard.h.

at::cuda::CUDAMultiStreamGuard::~CUDAMultiStreamGuard ( )

inline

Resets the CUDA stream on each device to the one that was active upon construction.

Definition at line 47 of file CUDAMultiStreamGuard.h.

The documentation for this class was generated from the following file: