dgpu

Documentation

Index

• func NewForDevice(d device.Device, opts ...Option) (discover.Discover, error)
• func NewForMigDevice(d device.Device, mig device.MigDevice, opts ...Option) (discover.Discover, error)
• type Option
  • func WithDevRoot(root string) Option
  • func WithLogger(logger logger.Interface) Option
  • func WithMIGCaps(migCaps nvcaps.MigCaps) Option
  • func WithNVIDIACDIHookPath(path string) Option
  • func WithNvsandboxuitilsLib(nvsandboxutilslib nvsandboxutils.Interface) Option
• type UUIDer

Functions

func NewForDevice

func NewForDevice(d device.Device, opts ...Option) (discover.Discover, error)

NewForDevice creates a discoverer for the specified Device. nvsandboxutils is used for discovery if specified, otherwise NVML is used.

func NewForMigDevice

func NewForMigDevice(d device.Device, mig device.MigDevice, opts ...Option) (discover.Discover, error)

NewForMigDevice creates a discoverer for the specified device and its associated MIG device. nvsandboxutils is used for discovery if specified, otherwise NVML is used.

Types

type Option

type Option func(*options)

func WithDevRoot

func WithDevRoot(root string) Option

WithDevRoot sets the root where /dev is located.

func WithLogger

func WithLogger(logger logger.Interface) Option

WithLogger sets the logger for the library

func WithMIGCaps

func WithMIGCaps(migCaps nvcaps.MigCaps) Option

WithMIGCaps sets the MIG capabilities.

func WithNVIDIACDIHookPath

func WithNVIDIACDIHookPath(path string) Option

WithNVIDIACDIHookPath sets the path to the NVIDIA Container Toolkit CLI path for the library

func WithNvsandboxuitilsLib

func WithNvsandboxuitilsLib(nvsandboxutilslib nvsandboxutils.Interface) Option

WithNvsandboxuitilsLib sets the nvsandboxutils library implementation.

type UUIDer

type UUIDer interface {
	GetUUID() (string, nvml.Return)
}