dispatcher

Documentation

Index

• Constants
• func ModelCompleteNotification(modelMapper *ModelMapper, datahubGrpcCn *grpc.ClientConn, ...)
• func NewGPUModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper, ...) *gpuModelJobSender
• func NewModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper, ...) *modelJobSender
• func NewNodeModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper, ...) *nodeModelJobSender
• func NewPodModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper, ...) *podModelJobSender
• func NewPredictJobSender(datahubGrpcCn *grpc.ClientConn) *predictJobSender
• type Dispatcher
  • func NewDispatcher(datahubGrpcCn *grpc.ClientConn, granularities []string, predictUnits []string, ...) *Dispatcher
  • func (dispatcher *Dispatcher) Start()
• type ModelMapper
  • func NewModelMapper(predictUnitTypes []string, granularities []string) *ModelMapper
  • func (mm *ModelMapper) AddModelInfo(predictUnitType string, granularity string, mInfo *modelInfo)
  • func (mm *ModelMapper) GetModelInfo(predictUnitType string, granularity string, modelID string) *modelInfo
  • func (mm *ModelMapper) IsModelTimeout(predictUnitType string, granularity string, mInfo *modelInfo) bool
  • func (mm *ModelMapper) IsModeling(predictUnitType string, granularity string, mInfo *modelInfo) bool
  • func (mm *ModelMapper) RemoveModelInfo(predictUnitType string, granularity string, modelID string)

Constants

const (
	UnitTypeNode = "NODE"
	UnitTypePod  = "POD"
	UnitTypeGPU  = "GPU"
)

Functions

func ModelCompleteNotification

func ModelCompleteNotification(modelMapper *ModelMapper,
	datahubGrpcCn *grpc.ClientConn, metricExporter *metrics.Exporter)

func NewGPUModelJobSender

func NewGPUModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper,
	metricExporter *metrics.Exporter) *gpuModelJobSender

func NewModelJobSender

func NewModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper,
	metricExporter *metrics.Exporter) *modelJobSender

func NewNodeModelJobSender

func NewNodeModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper,
	metricExporter *metrics.Exporter) *nodeModelJobSender

func NewPodModelJobSender

func NewPodModelJobSender(datahubGrpcCn *grpc.ClientConn, modelMapper *ModelMapper,
	metricExporter *metrics.Exporter) *podModelJobSender

func NewPredictJobSender

func NewPredictJobSender(datahubGrpcCn *grpc.ClientConn) *predictJobSender

Types

type Dispatcher

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

func NewDispatcher

func NewDispatcher(datahubGrpcCn *grpc.ClientConn, granularities []string,
	predictUnits []string, modelMapper *ModelMapper, metricExporter *metrics.Exporter) *Dispatcher

func (*Dispatcher) Start

func (dispatcher *Dispatcher) Start()

type ModelMapper

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

func NewModelMapper

func NewModelMapper(predictUnitTypes []string, granularities []string) *ModelMapper

func (*ModelMapper) AddModelInfo

func (mm *ModelMapper) AddModelInfo(predictUnitType string,
	granularity string, mInfo *modelInfo)

func (*ModelMapper) GetModelInfo

func (mm *ModelMapper) GetModelInfo(predictUnitType string,
	granularity string, modelID string) *modelInfo

func (*ModelMapper) IsModelTimeout

func (mm *ModelMapper) IsModelTimeout(predictUnitType string,
	granularity string, mInfo *modelInfo) bool

func (*ModelMapper) IsModeling

func (mm *ModelMapper) IsModeling(predictUnitType string,
	granularity string, mInfo *modelInfo) bool

func (*ModelMapper) RemoveModelInfo

func (mm *ModelMapper) RemoveModelInfo(predictUnitType string,
	granularity string, modelID string)