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

update

Browse Source
This commit is contained in:
Newnius 2020-06-11 00:23:41 +08:00
parent 8734e56816
commit 01661876fd
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/resource_pool.go
View File

@ -140,7 +140,6 @@ func (pool *ResourcePool) init(conf Configuration) {
var tasks []Task var tasks []Task
for _, job := range pool.batchJobs { for _, job := range pool.batchJobs {
for _, task := range job.Tasks { for _, task := range job.Tasks {
task.Job = job.Name
tasks = append(tasks, task) tasks = append(tasks, task)
} }
} }
@ -693,10 +692,10 @@ func (pool *ResourcePool) pickNode(candidates []*NodeStatus, availableGPUs map[s
} }
func (pool *ResourcePool) acquireResource(job Job) []NodeStatus { func (pool *ResourcePool) acquireResource(job Job) []NodeStatus {
for i := range job.Tasks {
job.Tasks[i].Job = job.Name
}
if !pool.enableBatch { if !pool.enableBatch {
for i := range job.Tasks {
job.Tasks[i].Job = job.Name
}
return pool.doAcquireResource(job) return pool.doAcquireResource(job)
} }
pool.batchMu.Lock() pool.batchMu.Lock()