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 ...@@ -411,8 +411,6 @@ func (w *Worker) doRunningModelStatus(msg *omanager.WorkerMessage_RunningModelSt
//if !w.registed { //if !w.registed {
// return // return
//} //}
// todo: profile performance.
return
log.WithFields(log.Fields{ log.WithFields(log.Fields{
"worker-addr": w.workerAddr, "worker-addr": w.workerAddr,
}).Debugf("receive worker running model status:%v", msg.RunningModelStatus) }).Debugf("receive worker running model status:%v", msg.RunningModelStatus)
...@@ -431,8 +429,6 @@ func (w *Worker) doRunningModelStatus(msg *omanager.WorkerMessage_RunningModelSt ...@@ -431,8 +429,6 @@ func (w *Worker) doRunningModelStatus(msg *omanager.WorkerMessage_RunningModelSt
} }
func (w *Worker) doGPUUsage(msg *omanager.WorkerMessage_GpuUsage) { func (w *Worker) doGPUUsage(msg *omanager.WorkerMessage_GpuUsage) {
// todo: profile performance.
return
if w.usage.hwUsage == nil { if w.usage.hwUsage == nil {
return return
} }
......
...@@ -459,14 +459,16 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) { ...@@ -459,14 +459,16 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
case *omanager.WorkerMessage_NodeInfo: case *omanager.WorkerMessage_NodeInfo:
worker.doGetNodeInfo(msg) worker.doGetNodeInfo(msg)
case *omanager.WorkerMessage_FetchStandardTask: //case *omanager.WorkerMessage_FetchStandardTask:
worker.doFetchStdTask(msg) // worker.doFetchStdTask(msg)
case *omanager.WorkerMessage_DeviceInfo: //case *omanager.WorkerMessage_DeviceInfo:
worker.doGetDeviceInfo(msg) // worker.doGetDeviceInfo(msg)
//
case *omanager.WorkerMessage_DeviceUsage: //case *omanager.WorkerMessage_DeviceUsage:
worker.doDeviceUsage(msg) // worker.doDeviceUsage(msg)
//case *omanager.WorkerMessage_GpuUsage:
// worker.doGPUUsage(msg)
case *omanager.WorkerMessage_AddModelRunning: case *omanager.WorkerMessage_AddModelRunning:
worker.doAddRunningModel(msg) worker.doAddRunningModel(msg)
...@@ -486,9 +488,6 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) { ...@@ -486,9 +488,6 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
case *omanager.WorkerMessage_RunningModelStatus: case *omanager.WorkerMessage_RunningModelStatus:
worker.doRunningModelStatus(msg) worker.doRunningModelStatus(msg)
case *omanager.WorkerMessage_GpuUsage:
worker.doGPUUsage(msg)
case *omanager.WorkerMessage_RegisteMessage: case *omanager.WorkerMessage_RegisteMessage:
// 1. do some verify. // 1. do some verify.
if worker.registed { 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