gpuallocator

Nov 17, 2023

Changes in this version

+ type DeviceList []*Device

+ func (d DeviceList) Filter(uuids []string) (DeviceList, error)

+ type Option func(*deviceListBuilder)

+ func WithDeviceLib(devicelib device.Interface) Option

+ func WithNvmlLib(nvmllib nvml.Interface) Option

Jul 17, 2020

Changes in this version

type Device

+ func NewDevices() ([]*Device, error)

+ func NewDevicesFrom(uuids []string) ([]*Device, error)

type DeviceSet

+ func (ds DeviceSet) ContainsAll(devices []*Device) bool

Jun 6, 2019

Changes in this version

+ type Allocator struct

+ GPUs []*Device

+ func NewAllocator(policy Policy) (*Allocator, error)

+ func NewBestEffortAllocator() (*Allocator, error)

+ func NewSimpleAllocator() (*Allocator, error)

+ func (a *Allocator) Allocate(num int) []*Device

+ func (a *Allocator) AllocateSpecific(devices ...*Device) error

+ func (a *Allocator) Free(devices ...*Device)

+ type Device struct

+ Index int

+ Links map[int][]P2PLink

+ func (d *Device) Details() string

+ func (d *Device) String() string

+ type DeviceSet map[string]*Device

+ func NewDeviceSet() DeviceSet

+ func (ds DeviceSet) Contains(device *Device) bool

+ func (ds DeviceSet) Delete(devices ...*Device)

+ func (ds DeviceSet) Insert(devices ...*Device)

+ func (ds DeviceSet) SortedSlice() []*Device

+ type GPUType int

+ const GPUTypePascal

+ const GPUTypeVolta

+ type P2PLink struct

+ GPU *Device

+ Type nvml.P2PLinkType

+ type Policy interface

+ Allocate func(devices []*Device, size int) []*Device

+ func NewBestEffortPolicy() Policy

+ func NewSimplePolicy() Policy

+ func NewStaticDGX1Policy(gpuType GPUType) Policy

+ func NewStaticDGX2Policy() Policy