device

Mar 14, 2026

Changes in this version

type Allocator

+ func NewCUDAAllocator(deviceID int) Allocator

type Type

+ const OpenCL

+ const ROCm

Aug 4, 2025

Changes in this version

+ type Allocator interface

+ Allocate func(size int) (any, error)

+ Free func(ptr any) error

+ func NewCPUAllocator() Allocator

+ type Device interface

+ GetAllocator func() Allocator

+ ID func() string

+ Type func() Type

+ func Get(id string) (Device, error)

+ type Type int

+ const CPU

+ const CUDA