gpu

Documentation

Index

• func GetVendorByName(vendor string) (sonm.GPUVendorType, error)
• type DRICard
  • func CollectDRICardDevices() ([]DRICard, error)
  • func NewDRICard(num int, name, path string) (DRICard, error)
  • func (card *DRICard) Metrics() (*DRICardMetrics, error)
• type DRICardMetrics
• type GPUID
• type NilTuner
  • func (NilTuner) Close() error
  • func (NilTuner) Devices() []*sonm.GPUDevice
  • func (NilTuner) Tune(hostconfig *container.HostConfig, ids []GPUID) error
• type Option
  • func WithOptions(raw map[string]string) Option
  • func WithSocketDir(dir string) Option
• type Tuner
  • func New(ctx context.Context, gpuType sonm.GPUVendorType, opts ...Option) (Tuner, error)

Functions

func GetVendorByName

func GetVendorByName(vendor string) (sonm.GPUVendorType, error)

Types

type DRICard

type DRICard struct {
	Num       int
	Name      string
	Path      string
	Devices   []string
	DeviceID  uint64
	VendorID  uint64
	Major     uint64
	Minor     uint64
	PCIBusID  string
	Memory    uint64
	HwmonPath string
}

func CollectDRICardDevices

func CollectDRICardDevices() ([]DRICard, error)

CollectDRICardDevices traverses overs /dev/dri and collect card Devices which can be bound into the container

func NewDRICard

func NewDRICard(num int, name, path string) (DRICard, error)

func (*DRICard) Metrics

func (card *DRICard) Metrics() (*DRICardMetrics, error)

type DRICardMetrics

type DRICardMetrics struct {
	Temperature float64
	Fan         float64
	Power       float64
}

type GPUID

type GPUID string

type NilTuner

type NilTuner struct{}

NilTuner is just a null pattern

func (NilTuner) Close

func (NilTuner) Close() error

func (NilTuner) Devices

func (NilTuner) Devices() []*sonm.GPUDevice

func (NilTuner) Tune

func (NilTuner) Tune(hostconfig *container.HostConfig, ids []GPUID) error

type Option

type Option func(*tunerOptions)

func WithOptions

func WithOptions(raw map[string]string) Option

func WithSocketDir

func WithSocketDir(dir string) Option

type Tuner

type Tuner interface {
	// Tune is attaching GPUs with given IDs into a container
	Tune(hostconfig *container.HostConfig, ids []GPUID) error
	// Devices returns device list that this tuner can control
	Devices() []*sonm.GPUDevice
	// Close closes the tuner and removes related files and sockets from the host system
	Close() error
}

Tuner is responsible for preparing GPU-friendly environment and baking proper options in container.HostConfig

func New

func New(ctx context.Context, gpuType sonm.GPUVendorType, opts ...Option) (Tuner, error)

New creates Tuner based on provided config. Currently supported type: "radeon" or "nvidia". if type is undefined, nilTuner will be returned.