Expand description
GPU memory management
Efficient memory allocation and pooling for GPU tensors
Structs§
- GpuMemory
Pool - GPU memory pool for efficient allocation
- GpuTensor
- GPU tensor wrapper
Functions§
- get_
global_ gpu_ pool - Get or create the global GPU memory pool