mirror of
https://github.com/newnius/YAO-scheduler.git
synced 2025-12-12 23:36:44 +00:00
bugfix
This commit is contained in:
@@ -12,10 +12,11 @@ import (
|
||||
)
|
||||
|
||||
type JobManager struct {
|
||||
scheduler Scheduler
|
||||
job Job
|
||||
jobStatus JobStatus
|
||||
resources []NodeStatus
|
||||
scheduler Scheduler
|
||||
job Job
|
||||
jobStatus JobStatus
|
||||
resources []NodeStatus
|
||||
killedFlag bool
|
||||
}
|
||||
|
||||
func (jm *JobManager) start() {
|
||||
@@ -28,6 +29,9 @@ func (jm *JobManager) start() {
|
||||
for i := range jm.job.Tasks {
|
||||
var resource NodeStatus
|
||||
for {
|
||||
if jm.killedFlag {
|
||||
break
|
||||
}
|
||||
resource = jm.scheduler.AcquireResource(jm.job, jm.job.Tasks[i])
|
||||
if len(resource.Status) > 0 {
|
||||
break
|
||||
@@ -41,6 +45,9 @@ func (jm *JobManager) start() {
|
||||
|
||||
/* bring up containers */
|
||||
for i := range jm.job.Tasks {
|
||||
if jm.killedFlag {
|
||||
break
|
||||
}
|
||||
var GPUs []string
|
||||
for _, GPU := range jm.resources[i].Status {
|
||||
GPUs = append(GPUs, GPU.UUID)
|
||||
|
||||
@@ -33,10 +33,7 @@ func (s FairJobSorter) Swap(i, j int) {
|
||||
s[i], s[j] = s[j], s[i]
|
||||
}
|
||||
func (s FairJobSorter) Less(i, j int) bool {
|
||||
if s[i].Priority > s[j].Priority {
|
||||
return true
|
||||
}
|
||||
return s[i].CreatedAt > s[j].CreatedAt
|
||||
return s[i].CreatedAt < s[j].CreatedAt
|
||||
}
|
||||
|
||||
func (scheduler *SchedulerFair) Start() {
|
||||
@@ -362,7 +359,5 @@ func (scheduler *SchedulerFair) UpdateNextQueue() {
|
||||
}
|
||||
}
|
||||
scheduler.nextQueue = next
|
||||
log.Info("updateNextQueue")
|
||||
log.Info(scheduler.resourceAllocations)
|
||||
log.Info("updateNextQueue ->", next)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user