nvidia

Documentation

Index

• Constants
• Variables
• func ContainsSliceFunc[S ~[]E, E any](s S, match func(E) bool) bool
• func ParseConfig(fs *flag.FlagSet)
• type NvidiaGPUDevices
  • func InitNvidiaDevice() *NvidiaGPUDevices
  • func (dev *NvidiaGPUDevices) CheckHealth(devType string, n *corev1.Node) (bool, bool)
  • func (dev *NvidiaGPUDevices) CheckType(annos map[string]string, d util.DeviceUsage, n util.ContainerDeviceRequest) (bool, bool, bool)
  • func (dev *NvidiaGPUDevices) CheckUUID(annos map[string]string, d util.DeviceUsage) bool
  • func (dev *NvidiaGPUDevices) CommonWord() string
  • func (dev *NvidiaGPUDevices) GenerateResourceRequests(ctr *corev1.Container) util.ContainerDeviceRequest
  • func (dev *NvidiaGPUDevices) GetNodeDevices(n corev1.Node) ([]*api.DeviceInfo, error)
  • func (dev *NvidiaGPUDevices) LockNode(n *corev1.Node, p *corev1.Pod) error
  • func (dev *NvidiaGPUDevices) MutateAdmission(ctr *corev1.Container) (bool, error)
  • func (dev *NvidiaGPUDevices) NodeCleanUp(nn string) error
  • func (dev *NvidiaGPUDevices) PatchAnnotations(annoinput *map[string]string, pd util.PodDevices) map[string]string
  • func (dev *NvidiaGPUDevices) ReleaseNodeLock(n *corev1.Node, p *corev1.Pod) error

Constants

const (
	HandshakeAnnos      = "hami.io/node-handshake"
	RegisterAnnos       = "hami.io/node-nvidia-register"
	NvidiaGPUDevice     = "NVIDIA"
	NvidiaGPUCommonWord = "GPU"
	GPUInUse            = "nvidia.com/use-gputype"
	GPUNoUse            = "nvidia.com/nouse-gputype"
	NumaBind            = "nvidia.com/numa-bind"
	NodeLockNvidia      = "hami.io/mutex.lock"
	// GPUUseUUID is user can use specify GPU device for set GPU UUID.
	GPUUseUUID = "nvidia.com/use-gpuuuid"
	// GPUNoUseUUID is user can not use specify GPU device for set GPU UUID.
	GPUNoUseUUID = "nvidia.com/nouse-gpuuuid"
)

Variables

var (
	ResourceName          string
	ResourceMem           string
	ResourceCores         string
	ResourceMemPercentage string
	ResourcePriority      string
	DebugMode             bool
	OverwriteEnv          bool
)

Functions

func ContainsSliceFunc

func ContainsSliceFunc[S ~[]E, E any](s S, match func(E) bool) bool

func ParseConfig

func ParseConfig(fs *flag.FlagSet)

Types

type NvidiaGPUDevices

type NvidiaGPUDevices struct {
}

func InitNvidiaDevice

func InitNvidiaDevice() *NvidiaGPUDevices

func (*NvidiaGPUDevices) CheckHealth

func (dev *NvidiaGPUDevices) CheckHealth(devType string, n *corev1.Node) (bool, bool)

func (*NvidiaGPUDevices) CheckType

func (dev *NvidiaGPUDevices) CheckType(annos map[string]string, d util.DeviceUsage, n util.ContainerDeviceRequest) (bool, bool, bool)

func (*NvidiaGPUDevices) CheckUUID

func (dev *NvidiaGPUDevices) CheckUUID(annos map[string]string, d util.DeviceUsage) bool

func (*NvidiaGPUDevices) CommonWord

func (dev *NvidiaGPUDevices) CommonWord() string

func (*NvidiaGPUDevices) GenerateResourceRequests

func (dev *NvidiaGPUDevices) GenerateResourceRequests(ctr *corev1.Container) util.ContainerDeviceRequest

func (*NvidiaGPUDevices) GetNodeDevices

func (dev *NvidiaGPUDevices) GetNodeDevices(n corev1.Node) ([]*api.DeviceInfo, error)

func (*NvidiaGPUDevices) LockNode

func (dev *NvidiaGPUDevices) LockNode(n *corev1.Node, p *corev1.Pod) error

func (*NvidiaGPUDevices) MutateAdmission

func (dev *NvidiaGPUDevices) MutateAdmission(ctr *corev1.Container) (bool, error)

func (*NvidiaGPUDevices) NodeCleanUp

func (dev *NvidiaGPUDevices) NodeCleanUp(nn string) error

func (*NvidiaGPUDevices) PatchAnnotations

func (dev *NvidiaGPUDevices) PatchAnnotations(annoinput *map[string]string, pd util.PodDevices) map[string]string

func (*NvidiaGPUDevices) ReleaseNodeLock

func (dev *NvidiaGPUDevices) ReleaseNodeLock(n *corev1.Node, p *corev1.Pod) error