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) } }