controller

Documentation

Index

• func StartLeaseControllerErrExit(ctx context.Context, kubeClient kubernetes.Interface, ...)
• type HostGpuInfoChecker
  • func NewHostGpuInfoChecker(checkInterval time.Duration, stop <-chan struct{}) (*HostGpuInfoChecker, error)
  • func (gic *HostGpuInfoChecker) GetGpuInfoChan() <-chan *NodeGpuInfo
  • func (gic *HostGpuInfoChecker) Start() error
• type PodWatcher
  • func NewPodWatcher(kubeclient kubernetes.Interface, stop <-chan struct{}) (*PodWatcher, error)
  • func (pw *PodWatcher) GetRemoveChan() <-chan *PodResourceUpdate
  • func (pw *PodWatcher) GetSyncChan() <-chan struct{}
  • func (pw *PodWatcher) Start() error
• type ServerDSController
  • func NewServerDSController(stop <-chan struct{}, goonChan <-chan struct{}, ...) (*ServerDSController, error)
  • func (dsc *ServerDSController) Start() error

Functions

func StartLeaseControllerErrExit

func StartLeaseControllerErrExit(ctx context.Context, kubeClient kubernetes.Interface, gpuClient gpuclientset.Interface)

Types

type HostGpuInfoChecker

type HostGpuInfoChecker struct {
	// contains filtered or unexported fields
}

HostGpuInfoChecker check node gpu model and populate it. It depends on the package github.com/NVIDIA/go-nvml/pkg/nvml It Start to populate gpu model info each interval and never stop until stop chan signal.

func NewHostGpuInfoChecker

func NewHostGpuInfoChecker(checkInterval time.Duration, stop <-chan struct{}) (*HostGpuInfoChecker, error)

func (*HostGpuInfoChecker) GetGpuInfoChan

func (gic *HostGpuInfoChecker) GetGpuInfoChan() <-chan *NodeGpuInfo

func (*HostGpuInfoChecker) Start

func (gic *HostGpuInfoChecker) Start() error

type PodWatcher

type PodWatcher struct {
	// contains filtered or unexported fields
}

PodWatcher watch the pod from of each node, filter by gpu and pod condition running.

func NewPodWatcher

func NewPodWatcher(kubeclient kubernetes.Interface, stop <-chan struct{}) (*PodWatcher, error)

func (*PodWatcher) GetRemoveChan

func (pw *PodWatcher) GetRemoveChan() <-chan *PodResourceUpdate

func (*PodWatcher) GetSyncChan

func (pw *PodWatcher) GetSyncChan() <-chan struct{}

func (*PodWatcher) Start

func (pw *PodWatcher) Start() error

type ServerDSController

type ServerDSController struct {
	// contains filtered or unexported fields
}

ServerDSController is the main controller to signal gpu usage info to the server. Signal pod gpu usage info to the server. Signal node gpu info to the server.

func NewServerDSController

func NewServerDSController(stop <-chan struct{}, goonChan <-chan struct{}, removeChan <-chan *PodResourceUpdate, gpuinfoChan <-chan *NodeGpuInfo, podresourcesep string, gpuClient gpuclientset.Interface, gpuPodClient gpupodcleintset.Interface) (*ServerDSController, error)

func (*ServerDSController) Start

func (dsc *ServerDSController) Start() error