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

---
 src/scheduler_fair.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go
index fe55708..57e30be 100644
--- a/src/scheduler_fair.go
+++ b/src/scheduler_fair.go
@@ -257,10 +257,17 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) NodeStatus {
 
 	/*assign*/
 	if len(candidates) > 0 {
+		var available []GPUStatus
+		for _, status := range candidates[0].Status {
+			if status.MemoryTotal >= task.MemoryGPU && status.MemoryUsed < 10 {
+				available = append(available, status)
+			}
+		}
+
 		node := candidates[0]
 		res.ClientID = node.ClientID
 		res.ClientHost = node.ClientHost
-		res.Status = candidates[0].Status[0:task.NumberGPU]
+		res.Status = available[0:task.NumberGPU]
 		res.NumCPU = task.NumberCPU
 		res.MemTotal = task.Memory