Module gpu

Module gpu 

Source
Expand description

GPU Execution Backend (Stub for Future Integration)

This module provides a framework for GPU-accelerated computation graph execution. Currently implements stubs that will be filled in when CUDA/OpenCL/Vulkan integration is added.

§Future Integration Points

  • CUDA support via cuda-sys or cudarc
  • OpenCL support via opencl3
  • Vulkan compute support via vulkano or ash
  • Metal support for Apple Silicon
  • ROCm support for AMD GPUs

Structs§

GpuBuffer
GPU buffer for storing tensor data
GpuDevice
GPU device information
GpuExecutor
GPU executor for computation graphs
GpuKernel
GPU kernel for executing operations
GpuMemoryManager
GPU memory manager for optimal allocation

Enums§

GpuBackend
GPU device types
GpuError
Errors that can occur during GPU operations