mgsrv

Documentation

Index

• type MetaGpuServer
  • func NewMetaGpuServer(_logger *zap.Logger, nodeName string, enforceKill bool, serverAddr string, ...) *MetaGpuServer
  • func (s *MetaGpuServer) GenerateAuthTokens(visibility VisibilityLevel) string
  • func (s *MetaGpuServer) IsMethodPublic(fullMethod string) bool
  • func (s *MetaGpuServer) Start()
• type MetaGpuServerStream
  • func (s *MetaGpuServerStream) Context() context.Context
• type VisibilityLevel

Types

type MetaGpuServer

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

func NewMetaGpuServer

func NewMetaGpuServer(_logger *zap.Logger,
	nodeName string,
	enforceKill bool,
	serverAddr string,
	jwtSecret string,
	shareConf *sharecfg.DevicesSharingConfigs,
) *MetaGpuServer

NewMetaGpuServer - creates a new server instance. nodeName - hostname of a worker on which this instance is running as it is in Kube API. enforceKill - if set, enables GPU OOMKill.

func (*MetaGpuServer) GenerateAuthTokens

func (s *MetaGpuServer) GenerateAuthTokens(visibility VisibilityLevel) string

func (*MetaGpuServer) IsMethodPublic

func (s *MetaGpuServer) IsMethodPublic(fullMethod string) bool

func (*MetaGpuServer) Start

func (s *MetaGpuServer) Start()

type MetaGpuServerStream

type MetaGpuServerStream struct {
	grpc.ServerStream
	// contains filtered or unexported fields
}

func (*MetaGpuServerStream) Context

func (s *MetaGpuServerStream) Context() context.Context

type VisibilityLevel

type VisibilityLevel string

const (
	DeviceVisibility         VisibilityLevel = "l0"
	ContainerVisibility      VisibilityLevel = "l1"
	TokenVisibilityClaimName                 = "visibilityLevel"
)