API Reference

// Set the compute backend (Auto, CPU, Metal, CUDA, WebGPU)
func SetBackend(backend Backend)

// Get the current backend
func GetBackend() Backend

// Get device info (name, vendor, memory, compute units)
func GetDevice() Device

// Get runtime info string
func Info() string

// Wait for all pending GPU operations to complete
func Synchronize()

type Array struct {
    // opaque - use methods below
}

func (a *Array) Shape() []int   // dimensions
func (a *Array) Dtype() Dtype   // element type
func (a *Array) Size() int      // total element count
func (a *Array) Data() []float32 // read back to CPU (CPU backend only)

type Device struct {
    Type         Backend
    Index        int
    Name         string
    Vendor       string
    MemoryTotal  uint64
    MemoryAvail  uint64
    IsDiscrete   bool
    IsUnified    bool
    ComputeUnits int
    MaxWorkgroup int
}

var (
    ErrNoGPU          // no GPU available
    ErrInvalidBackend // invalid backend selection
    ErrOutOfMemory    // GPU out of memory
    ErrKernelFailed   // kernel execution failed
    ErrNotSupported   // operation not supported on backend
)

CGO_CFLAGS="-I/path/to/luxcpp/gpu/include"
CGO_LDFLAGS="-L/path/to/luxcpp/gpu/build -llux-gpu"
CGO_ENABLED=1 go build ./...