gpu-feature-discovery

Feb 4, 2021

Changes in this version

type Conf

+ NoTimestamp bool

+ type MIGCapableDevices struct

+ func NewMIGCapableDevices(nvml Nvml) *MIGCapableDevices

+ func (devices *MIGCapableDevices) AssertAllMigEnabledDevicesAreValid() error

+ func (devices *MIGCapableDevices) GetAllMigDevices() ([]NvmlDevice, error)

+ func (devices *MIGCapableDevices) GetDevicesWithMigDisabled() ([]NvmlDevice, error)

+ func (devices *MIGCapableDevices) GetDevicesWithMigEnabled() ([]NvmlDevice, error)

Jan 12, 2021

Changes in this version

+ const HostDriverBranchLength

+ const HostDriverVersionLength

+ const PciCapabilityLength

+ const PciCapabilityList

+ const PciCapabilityListID

+ const PciCapabilityListNext

+ const PciCapabilityVendorSpecificID

+ const PciDevicesRoot

+ const PciNvidiaVendorID

+ const PciStatusByte

+ const PciStatusCapabilityList

+ const VGPUCapabilityRecordStart

+ func GetByte(buffer []byte, pos int) uint8

+ func GetLong(buffer []byte, pos int) uint32

+ func GetWord(buffer []byte, pos int) uint16

type Conf

+ FailOnInitError bool

+ type NvidiaPCI interface

+ Devices func() ([]*PCIDevice, error)

+ func NewNvidiaPCILib() NvidiaPCI

+ type NvidiaPCILib struct

+ func (p *NvidiaPCILib) Devices() ([]*PCIDevice, error)

+ type NvmlInitError struct

+ type PCIDevice struct

+ Address string

+ Class string

+ Config []byte

+ Path string

+ Vendor string

+ func (d *PCIDevice) GetVendorSpecificCapability() ([]byte, error)

+ type VGPU interface

+ Devices func() ([]*VGPUDevice, error)

+ func NewVGPULib(pci NvidiaPCI) VGPU

+ type VGPUDevice struct

+ func (d *VGPUDevice) GetInfo() (*VGPUInfo, error)

+ type VGPUInfo struct

+ HostDriverBranch string

+ HostDriverVersion string

+ type VGPULib struct

+ func (v *VGPULib) Devices() ([]*VGPUDevice, error)

+ func (v *VGPULib) IsVGPUDevice(capability []byte) bool

Sep 26, 2020

Changes in this version

+ const MigStrategyMixed

+ const MigStrategyNone

+ const MigStrategySingle

type Conf

+ MigStrategy string

+ type MigDeviceCounts map[string]int

+ type MigStrategy interface

+ GenerateLabels func() (map[string]string, error)

+ func NewMigStrategy(strategy string, nvml Nvml) (MigStrategy, error)

+ type Nvml interface

+ GetCudaDriverVersion func() (*uint, *uint, error)

+ GetDeviceCount func() (uint, error)

+ GetDriverVersion func() (string, error)

+ Init func() error

+ NewDevice func(id uint) (device NvmlDevice, err error)

+ Shutdown func() error

+ type NvmlDevice interface

+ GetAttributes func() (nvml.DeviceAttributes, error)

+ GetMigDevices func() ([]NvmlDevice, error)

+ Instance func() *nvml.Device

+ IsMigEnabled func() (bool, error)

+ type NvmlLibDevice struct

+ func (d NvmlLibDevice) GetAttributes() (nvml.DeviceAttributes, error)

+ func (d NvmlLibDevice) GetMigDevices() ([]NvmlDevice, error)

+ func (d NvmlLibDevice) Instance() *nvml.Device

+ func (d NvmlLibDevice) IsMigEnabled() (bool, error)

+ type NvmlMockDevice struct

+ func (d NvmlMockDevice) GetAttributes() (nvml.DeviceAttributes, error)

+ func (d NvmlMockDevice) GetMigDevices() ([]NvmlDevice, error)

+ func (d NvmlMockDevice) Instance() *nvml.Device

+ func (d NvmlMockDevice) IsMigEnabled() (bool, error)

Aug 28, 2019

Changes in this version

+ const Bin

+ var MachineTypePath = "/sys/class/dmi/id/product_name"

+ var Version = ""

+ type Conf struct

+ Oneshot bool

+ OutputFilePath string

+ SleepInterval time.Duration

+ type NvmlInterface interface

+ GetCudaDriverVersion func() (*uint, *uint, error)

+ GetDeviceCount func() (uint, error)

+ GetDriverVersion func() (string, error)

+ Init func() error

+ NewDevice func(id uint) (device *nvml.Device, err error)

+ Shutdown func() error

+ type NvmlLib struct

+ func (nvmlLib NvmlLib) GetCudaDriverVersion() (*uint, *uint, error)

+ func (nvmlLib NvmlLib) GetDeviceCount() (uint, error)

+ func (nvmlLib NvmlLib) GetDriverVersion() (string, error)

+ func (nvmlLib NvmlLib) Init() error

+ func (nvmlLib NvmlLib) NewDevice(id uint) (device *nvml.Device, err error)

+ func (nvmlLib NvmlLib) Shutdown() error

+ type NvmlMock struct

+ func (nvmlMock NvmlMock) GetCudaDriverVersion() (*uint, *uint, error)

+ func (nvmlMock NvmlMock) GetDeviceCount() (uint, error)

+ func (nvmlMock NvmlMock) GetDriverVersion() (string, error)

+ func (nvmlMock NvmlMock) Init() error

+ func (nvmlMock NvmlMock) NewDevice(id uint) (*nvml.Device, error)

+ func (nvmlMock NvmlMock) Shutdown() error