Shortcuts

Struct CUDAMultiStreamGuard

Page Contents

Struct Documentation

struct CUDAMultiStreamGuard

A variant of MultiStreamGuard that is specialized for CUDA.

Public Functions

inline explicit CUDAMultiStreamGuard(ArrayRef<CUDAStream> streams)
CUDAMultiStreamGuard(const CUDAMultiStreamGuard&) = delete

Copy is disallowed.

CUDAMultiStreamGuard &operator=(const CUDAMultiStreamGuard&) = delete
CUDAMultiStreamGuard(CUDAMultiStreamGuard &&other) = delete
CUDAMultiStreamGuard &operator=(CUDAMultiStreamGuard &&other) = delete
~CUDAMultiStreamGuard() = default

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources