Commit e005f062 authored by vicotor's avatar vicotor

update for benchmark

parent e95def3b
......@@ -411,8 +411,6 @@ func (w *Worker) doRunningModelStatus(msg *omanager.WorkerMessage_RunningModelSt
//if !w.registed {
// return
//}
// todo: profile performance.
return
log.WithFields(log.Fields{
"worker-addr": w.workerAddr,
}).Debugf("receive worker running model status:%v", msg.RunningModelStatus)
......@@ -431,8 +429,6 @@ func (w *Worker) doRunningModelStatus(msg *omanager.WorkerMessage_RunningModelSt
}
func (w *Worker) doGPUUsage(msg *omanager.WorkerMessage_GpuUsage) {
// todo: profile performance.
return
if w.usage.hwUsage == nil {
return
}
......
......@@ -459,14 +459,16 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
case *omanager.WorkerMessage_NodeInfo:
worker.doGetNodeInfo(msg)
case *omanager.WorkerMessage_FetchStandardTask:
worker.doFetchStdTask(msg)
//case *omanager.WorkerMessage_FetchStandardTask:
// worker.doFetchStdTask(msg)
case *omanager.WorkerMessage_DeviceInfo:
worker.doGetDeviceInfo(msg)
case *omanager.WorkerMessage_DeviceUsage:
worker.doDeviceUsage(msg)
//case *omanager.WorkerMessage_DeviceInfo:
// worker.doGetDeviceInfo(msg)
//
//case *omanager.WorkerMessage_DeviceUsage:
// worker.doDeviceUsage(msg)
//case *omanager.WorkerMessage_GpuUsage:
// worker.doGPUUsage(msg)
case *omanager.WorkerMessage_AddModelRunning:
worker.doAddRunningModel(msg)
......@@ -486,9 +488,6 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
case *omanager.WorkerMessage_RunningModelStatus:
worker.doRunningModelStatus(msg)
case *omanager.WorkerMessage_GpuUsage:
worker.doGPUUsage(msg)
case *omanager.WorkerMessage_RegisteMessage:
// 1. do some verify.
if worker.registed {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment