Expand description
CUDA backend implementation for GPU operations
This module provides CUDA-specific implementations for GPU operations.
Structs§
- Cuda
Context - CUDA context wrapper
- Cuda
Operations - High-level CUDA operations wrapper
- Cuda
Stream - CUDA stream for asynchronous operations
- Memory
Stats - Memory usage statistics
Functions§
- get_
optimizer_ kernels - Get precompiled optimizer kernels