lm

Documentation

Index

• Constants
• type Labeler
  • func Merge(labelers ...Labeler) Labeler
  • func NewGPUResourceLabeler(config *spec.Config, device resource.Device, count int) (Labeler, error)
  • func NewGPUResourceLabelerWithoutSharing(device resource.Device, count int) (Labeler, error)
  • func NewLabelers(manager resource.Manager, vgpu vgpu.Interface, config *spec.Config) (Labeler, error)
  • func NewMIGResourceLabeler(resourceName spec.ResourceName, config *spec.Config, device resource.Device, ...) (Labeler, error)
  • func NewNVMLLabeler(manager resource.Manager, config *spec.Config) (Labeler, error)
  • func NewResourceLabeler(manager resource.Manager, config *spec.Config) (Labeler, error)
  • func NewTimestampLabeler(config *spec.Config) Labeler
  • func NewVGPULabeler(vgpu vgpu.Interface) Labeler
• type Labels
  • func (labels Labels) Labels() (Labels, error)
  • func (labels Labels) WriteTo(output io.Writer) (int64, error)
  • func (labels Labels) WriteToFile(path string) error
• type MigDeviceCounts

Constants

const (
	MigStrategyNone   = "none"
	MigStrategySingle = "single"
	MigStrategyMixed  = "mixed"
)

Constants representing different MIG strategies.

Types

type Labeler

type Labeler interface {
	Labels() (Labels, error)
}

Labeler defines an interface for generating labels

func Merge

func Merge(labelers ...Labeler) Labeler

Merge converts a set of labelers to a single composite labeler.

func NewGPUResourceLabeler

func NewGPUResourceLabeler(config *spec.Config, device resource.Device, count int) (Labeler, error)

NewGPUResourceLabeler creates a resource labeler for the specified full GPU device with the specified count

func NewGPUResourceLabelerWithoutSharing

func NewGPUResourceLabelerWithoutSharing(device resource.Device, count int) (Labeler, error)

NewGPUResourceLabelerWithoutSharing creates a resource labeler for the specified device that does not apply sharing labels.

func NewLabelers

func NewLabelers(manager resource.Manager, vgpu vgpu.Interface, config *spec.Config) (Labeler, error)

NewLabelers constructs the required labelers from the specified config

func NewMIGResourceLabeler

func NewMIGResourceLabeler(resourceName spec.ResourceName, config *spec.Config, device resource.Device, count int) (Labeler, error)

NewMIGResourceLabeler creates a resource labeler for the specified full GPU device with the specified resource name.

func NewNVMLLabeler

func NewNVMLLabeler(manager resource.Manager, config *spec.Config) (Labeler, error)

NewNVMLLabeler creates a new NVML-based labeler using the provided NVML library and config.

func NewResourceLabeler

func NewResourceLabeler(manager resource.Manager, config *spec.Config) (Labeler, error)

NewResourceLabeler creates a labeler for available GPU resources. These include full GPU labels as well as labels specific to the mig-strategy specified.

func NewTimestampLabeler

func NewTimestampLabeler(config *spec.Config) Labeler

NewTimestampLabeler creates a new label manager for generating timestamp labels from the specified config. If the noTimestamp option is set an empty label manager is returned.

func NewVGPULabeler

func NewVGPULabeler(vgpu vgpu.Interface) Labeler

NewVGPULabeler creates a new VGP label manager using the provided vgpu library and config.

type Labels

type Labels map[string]string

Labels defines a type for labels

func (Labels) Labels

func (labels Labels) Labels() (Labels, error)

Labels also implements the Labeler interface

func (Labels) WriteTo

func (labels Labels) WriteTo(output io.Writer) (int64, error)

WriteTo writes labels to the specified writer

func (Labels) WriteToFile

func (labels Labels) WriteToFile(path string) error

WriteToFile writes labels to the specified path. The file is written atomocally

type MigDeviceCounts

type MigDeviceCounts map[string]int

MigDeviceCounts maintains a count of unique MIG device types across all GPUs on a node