YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-06 22:01:55 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-07-06 10:29:22 +08:00
parent aac4eefaef
commit a8045aaefb
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/job_manager.go
View File

@ -88,7 +88,7 @@ func (jm *JobManager) start() {
if GPU.MemoryUsed == GPU.MemoryTotal { if GPU.MemoryUsed == GPU.MemoryTotal {
shouldWait = "1" shouldWait = "1"
isScheduleAhead = true isScheduleAhead = true
} else if GPU.MemoryUsed > 0 { } else if GPU.MemoryAllocated > 0 {
isShare = true isShare = true
} }
/* attach to GPUs */ /* attach to GPUs */

6
src/resource_pool.go
View File

@ -714,7 +714,6 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
if len(job.Tasks) == 0 { if len(job.Tasks) == 0 {
return []NodeStatus{} return []NodeStatus{}
} }
task := job.Tasks[0]
segID := rand.Intn(pool.poolsCount) segID := rand.Intn(pool.poolsCount)
if pool.TotalGPU < 100 { if pool.TotalGPU < 100 {
segID = 0 segID = 0
@ -739,6 +738,7 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
loadRatio := float64(pool.UsingGPU) / float64(pool.TotalGPU) loadRatio := float64(pool.UsingGPU) / float64(pool.TotalGPU)
/* first, choose sharable GPUs */ /* first, choose sharable GPUs */
task := job.Tasks[0]
if len(job.Tasks) == 1 && task.NumberGPU == 1 && loadRatio >= config.EnableShareRatio { if len(job.Tasks) == 1 && task.NumberGPU == 1 && loadRatio >= config.EnableShareRatio {
// check sharable // check sharable
allocationType = 1 allocationType = 1
@ -803,8 +803,6 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
} }
node.Status[j].MemoryAllocated += task.MemoryGPU node.Status[j].MemoryAllocated += task.MemoryGPU
res.Status[i].MemoryTotal = task.MemoryGPU res.Status[i].MemoryTotal = task.MemoryGPU
/* being used, means share */
res.Status[i].MemoryUsed = 100
} }
} }
} }
@ -917,7 +915,7 @@ func (pool *ResourcePool) doAcquireResource(job Job) []NodeStatus {
node.Status[j].MemoryAllocated += task.MemoryGPU node.Status[j].MemoryAllocated += task.MemoryGPU
res.Status[i].MemoryTotal = task.MemoryGPU res.Status[i].MemoryTotal = task.MemoryGPU
/* being fully used, means ahead */ /* being fully used, means ahead */
res.Status[i].MemoryUsed = task.MemoryGPU res.Status[i].MemoryUsed = res.Status[i].MemoryTotal
} }
} }
} }