torch_sys_cuda

Module cuda

Source
Expand description

Bindings for torch’s wrappers around CUDA-related functionality.

Structs§

Event
Wrapper around a CUDA event.
Stream
Wrapper around a CUDA stream.

Enums§

CudaError
Corresponds to the CUDA error codes.

Functions§

cuda_check
set_device