CUDAGuard.h File Reference

#include <c10/core/DeviceType.h>
#include <c10/core/impl/InlineDeviceGuard.h>
#include <c10/core/impl/InlineStreamGuard.h>
#include <c10/cuda/CUDAMacros.h>
#include <c10/cuda/impl/CUDAGuardImpl.h>
#include <cstddef>

Go to the source code of this file.

Classes
struct	c10::cuda::CUDAGuard
	A variant of DeviceGuard that is specialized for CUDA. More...
struct	c10::cuda::OptionalCUDAGuard
	A variant of OptionalDeviceGuard that is specialized for CUDA. More...
struct	c10::cuda::CUDAStreamGuard
	A variant of StreamGuard that is specialized for CUDA. More...
struct	c10::cuda::OptionalCUDAStreamGuard
	A variant of OptionalStreamGuard that is specialized for CUDA. More...
struct	c10::cuda::CUDAMultiStreamGuard
	A variant of MultiStreamGuard that is specialized for CUDA. More...

Namespaces
namespace	c10
namespace	c10::cuda