From 420a1263bc276e94712cae7e57de88ddf7e907c7 Mon Sep 17 00:00:00 2001 From: Newnius Date: Thu, 30 Apr 2020 19:22:23 +0800 Subject: [PATCH] update --- src/scheduler_fair.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go index 4a43774..0fa03c5 100644 --- a/src/scheduler_fair.go +++ b/src/scheduler_fair.go @@ -350,6 +350,9 @@ func (scheduler *SchedulerFair) AcquireResource(job Job, task Task) NodeStatus { } } } + if allocationType == 2 { + scheduler.UsingGPU += task.NumberGPU + } } for i := range locks { @@ -392,6 +395,9 @@ func (scheduler *SchedulerFair) ReleaseResource(job Job, agent NodeStatus) { log.Warn(node.ClientID, "More Memory Allocated") node.Status[j].MemoryAllocated = 0 } + if node.Status[j].MemoryAllocated == 0 { + scheduler.UsingGPU-- + } log.Info(node.Status[j].MemoryAllocated) } }