Expand description
Real CUDA GPU backend using AvarokRegistry.
SBIO IORouter: all CUDA operations flow through GpuBackend.
Uses AvarokRegistry for kernel loading/launching and raw CUDA
driver API for memory management.
Modules§
- arch_
preflight - Refuse to load kernels the GPU cannot run, BEFORE the driver does it badly.
- tensormap
- TMA descriptors (
CUtensorMap) forcp.async.bulk.tensorloads.
Structs§
- Avarok
Cuda Backend - Production GPU backend wrapping AvarokRegistry + raw CUDA driver API.
Functions§
- cuda_
free_ memory_ bytes - Query GPU free memory without requiring a GpuBackend reference. Safe to call from any thread that shares the CUDA context.
- spawn_
oom_ watchdog - Start the OOM watchdog, or return
Noneif one is already running.