Module cuda

Module cuda 

Source
Expand description

A companion to the torch-sys crate that provides bindings for CUDA-specific functionality from libtorch. This crate is separated out to make it easier for clients who want to avoid compiling CUDA code.

The same safety logic described in the torch-sys crate applies here. 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