update

README.md

@@ -91,3 +91,13 @@ GPU is occupied by which job(s)
 ```
 ?action=debug_optimizer_describe_job&job=
 ```
+
+**EnableBatchAllocation**
+```
+?action=pool_enable_batch
+```
+
+**DisableBatchAllocation**
+```
+?action=pool_disable_batch
+```

src/main.go

@@ -295,6 +295,20 @@ func serverAPI(w http.ResponseWriter, r *http.Request) {
 		w.Write(js)
 		break
 
+	case "pool_enable_batch":
+		log.Debug("pool_enable_batch")
+		js, _ := json.Marshal(InstanceOfResourcePool().EnableBatch())
+		w.Header().Set("Content-Type", "application/json")
+		w.Write(js)
+		break
+
+	case "pool_disable_batch":
+		log.Debug("pool_disable_batch")
+		js, _ := json.Marshal(InstanceOfResourcePool().DisableBatch())
+		w.Header().Set("Content-Type", "application/json")
+		w.Write(js)
+		break
+
 	default:
 		http.Error(w, "Not Found", http.StatusNotFound)
 		break

src/resource_pool.go

@@ -124,7 +124,12 @@ func (pool *ResourcePool) init(conf Configuration) {
 	}()
 
 	go func() {
-		time.Sleep(time.Second * 10)
+		/* batch allocation */
+		for {
+			time.Sleep(time.Second * 15)
+			if !pool.enableBatch {
+				continue
+			}
 			pool.batchMu.Lock()
 			var tasks []Task
 			for _, job := range pool.batchJobs {
@@ -133,6 +138,7 @@ func (pool *ResourcePool) init(conf Configuration) {
 					tasks = append(tasks, task)
 				}
 			}
+			if len(tasks) != 0 {
 				job := Job{Tasks: tasks}
 
 				nodes := pool.doAcquireResource(job)
@@ -145,7 +151,9 @@ func (pool *ResourcePool) init(conf Configuration) {
 				if len(nodes) > 0 {
 					pool.batchJobs = []Job{}
 				}
+			}
 			pool.batchMu.Unlock()
+		}
 	}()
 }
 
@@ -885,9 +893,9 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
 			}
 		}
 
-		log.Info(tasks, factor)
+		//log.Info(tasks, factor)
 		allocation := InstanceOfAllocator().allocate(nodesT, tasks)
-		log.Info(allocation)
+		//log.Info(allocation)
 		if allocation.Flags["valid"] {
 
 			for range job.Tasks { //append would cause uncertain order
@@ -991,6 +999,18 @@ func (pool *ResourcePool) releaseResource(job Job, agent NodeStatus) {
 	}
 }
 
+func (pool *ResourcePool) EnableBatch() bool {
+	pool.enableBatch = true
+	log.Info("enableBatch is set to true")
+	return true
+}
+
+func (pool *ResourcePool) DisableBatch() bool {
+	pool.enableBatch = false
+	log.Info("enableBatch is set to false")
+	return true
+}
+
 func (pool *ResourcePool) SetShareRatio(ratio float64) bool {
 	pool.enableShareRatio = ratio
 	log.Info("enableShareRatio is updated to ", ratio)