From bbfd95f922ab90da65a3560fee462933d2f5801e Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Thu, 30 Apr 2020 17:11:16 +0800
Subject: [PATCH] update

---
 src/optimizer.go | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/optimizer.go b/src/optimizer.go
index e5f5b29..1102f19 100644
--- a/src/optimizer.go
+++ b/src/optimizer.go
@@ -48,14 +48,13 @@ func (optimizer *Optimizer) feed(job string, utils []int) {
 			for i := 0; i < len(utils); i++ {
 				sum += utils[i]
 			}
-			last := 0
-			version := 0
-			if t, err := optimizer.jobUtilsGPU[jobName]; !err {
-				last = t.Util
-				version = t.Version
+			sum /= len(utils)
+			if _, ok := optimizer.jobUtilsGPU[jobName]; !ok {
+				optimizer.jobUtilsGPU[jobName] = &OptimizerUtilGPU{}
 			}
-			optimizer.jobUtilsGPU[jobName].Util = (version*last + sum/len(utils)) / (version + 1)
-			optimizer.jobUtilsGPU[jobName].Version++
+			t := optimizer.jobUtilsGPU[jobName]
+			t.Util = (t.Version*t.Util + sum) / (t.Version + 1)
+			t.Version++
 
 			for i := 0; i < len(utils); i++ {
 				if utils[i] > 15 {