manager

package

v0.4.1 Latest Latest Go to latest Published: Apr 21, 2025 License: Apache-2.0 Imports: 23 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Device struct {
	GPU *GPUDevice
	MIG *MIGDevice
}

type DeviceManager struct {
	*nvidia.DeviceLib
	// contains filtered or unexported fields
}

func NewDeviceManager(config *node.NodeConfig, kubeClient *kubernetes.Clientset) (*DeviceManager, error)

func NewFakeDeviceManager(config *node.NodeConfig, version nvidia.DriverVersion, devices []*Device) *DeviceManager

func (m *DeviceManager) AddNotifyChannel(name string, ch chan *Device)

func (m *DeviceManager) GetDeviceInfoMap() map[string]device.DeviceInfo

func (m *DeviceManager) GetDriverVersion() nvidia.DriverVersion

func (m *DeviceManager) GetGPUDeviceMap() map[string]GPUDevice

func (m *DeviceManager) GetMIGDeviceMap() map[string]MIGDevice

func (m *DeviceManager) GetNodeConfig() node.NodeConfig

func (m *DeviceManager) GetNodeDeviceInfo() device.NodeDeviceInfo

func (m *DeviceManager) RemoveNotifyChannel(name string)

func (m *DeviceManager) Start()

func (m *DeviceManager) Stop()

type GPUDevice struct {
	*nvidia.GpuInfo
	NumaNode int
	Paths    []string
	Healthy  bool
	Links    map[int][]links.P2PLinkType
}

type MIGDevice struct {
	*nvidia.MigInfo
	Paths   []string
	Healthy bool
}