update

server/worker.go

@@ -444,7 +444,7 @@ func (w *Worker) doGPUUsage(msg *omanager.WorkerMessage_GpuUsage) {
 	log.WithFields(log.Fields{
 		"worker-addr": w.workerAddr,
 		"gpu-usage":   msg.GpuUsage,
-	}).Debugf("receive worker gpu usage")
+	}).Debug("receive worker gpu usage")
 	// 1. update gpu usage to hardware mogo.
 	gpuusages := make([]types.GpuUsage, 0)
 	for _, gpu := range msg.GpuUsage.Usages {

server/workerManager.go

@@ -460,16 +460,16 @@ func (wm *WorkerManager) handleWorkerMsg(worker *Worker) {
 			case *omanager.WorkerMessage_NodeInfo:
 				worker.doGetNodeInfo(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_GpuUsage:
-			//	worker.doGPUUsage(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_GpuUsage:
+				worker.doGPUUsage(msg)
 
 			case *omanager.WorkerMessage_AddModelRunning:
 				worker.doAddRunningModel(msg)