nvidia

Documentation

Index

• func NewVirtualGPUManager(vGPUCount int) *vGPUManager
• type NvidiaDevicePlugin
  • func NewNvidiaDevicePlugin(vGPUCount int) *NvidiaDevicePlugin
  • func (m *NvidiaDevicePlugin) Allocate(ctx context.Context, reqs *pluginapi.AllocateRequest) (*pluginapi.AllocateResponse, error)
  • func (m *NvidiaDevicePlugin) GetDevicePluginOptions(context.Context, *pluginapi.Empty) (*pluginapi.DevicePluginOptions, error)
  • func (m *NvidiaDevicePlugin) ListAndWatch(e *pluginapi.Empty, s pluginapi.DevicePlugin_ListAndWatchServer) error
  • func (m *NvidiaDevicePlugin) PreStartContainer(context.Context, *pluginapi.PreStartContainerRequest) (*pluginapi.PreStartContainerResponse, error)
  • func (m *NvidiaDevicePlugin) Register(kubeletEndpoint, resourceName string) error
  • func (m *NvidiaDevicePlugin) Serve() error
  • func (m *NvidiaDevicePlugin) Start() error
  • func (m *NvidiaDevicePlugin) Stop() error

Functions

func NewVirtualGPUManager

func NewVirtualGPUManager(vGPUCount int) *vGPUManager

NewVirtualGPUManager create a instance of vGPUManager

Types

type NvidiaDevicePlugin

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

NvidiaDevicePlugin implements the Kubernetes device plugin API

func NewNvidiaDevicePlugin

func NewNvidiaDevicePlugin(vGPUCount int) *NvidiaDevicePlugin

NewNvidiaDevicePlugin returns an initialized NvidiaDevicePlugin

func (*NvidiaDevicePlugin) Allocate

func (m *NvidiaDevicePlugin) Allocate(ctx context.Context, reqs *pluginapi.AllocateRequest) (*pluginapi.AllocateResponse, error)

Allocate which return list of devices.

func (*NvidiaDevicePlugin) GetDevicePluginOptions

func (m *NvidiaDevicePlugin) GetDevicePluginOptions(context.Context, *pluginapi.Empty) (*pluginapi.DevicePluginOptions, error)

func (*NvidiaDevicePlugin) ListAndWatch

func (m *NvidiaDevicePlugin) ListAndWatch(e *pluginapi.Empty, s pluginapi.DevicePlugin_ListAndWatchServer) error

ListAndWatch lists devices and update that list according to the health status

func (*NvidiaDevicePlugin) PreStartContainer

func (m *NvidiaDevicePlugin) PreStartContainer(context.Context, *pluginapi.PreStartContainerRequest) (*pluginapi.PreStartContainerResponse, error)

func (*NvidiaDevicePlugin) Register

func (m *NvidiaDevicePlugin) Register(kubeletEndpoint, resourceName string) error

Register registers the device plugin for the given resourceName with Kubelet.

func (*NvidiaDevicePlugin) Serve

func (m *NvidiaDevicePlugin) Serve() error

Serve starts the gRPC server and register the device plugin to Kubelet

func (*NvidiaDevicePlugin) Start

func (m *NvidiaDevicePlugin) Start() error

Start starts the gRPC server of the device plugin

func (*NvidiaDevicePlugin) Stop

func (m *NvidiaDevicePlugin) Stop() error

Stop stops the gRPC server