From 526ff91101ac86c00675537a6558112265687d0a Mon Sep 17 00:00:00 2001 From: JamesMurkin Date: Fri, 25 Oct 2024 10:43:42 +0100 Subject: [PATCH] Fix PreemptedAtPriority in fairshare preemption Signed-off-by: JamesMurkin --- internal/scheduler/nodedb/nodedb.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internal/scheduler/nodedb/nodedb.go b/internal/scheduler/nodedb/nodedb.go index 74c1af01c31..b2ab1676d6d 100644 --- a/internal/scheduler/nodedb/nodedb.go +++ b/internal/scheduler/nodedb/nodedb.go @@ -753,9 +753,9 @@ func (nodeDb *NodeDb) selectNodeForJobWithFairPreemption(txn *memdb.Txn, jctx *c return nil, errors.WithStack(err) } - priority, ok := nodeDb.GetScheduledAtPriority(evictedJctx.JobId) + priority, ok := nodeDb.GetScheduledAtPriority(job.JobSchedulingContext.JobId) if !ok { - priority = evictedJctx.Job.PriorityClass().Priority + priority = job.JobSchedulingContext.Job.PriorityClass().Priority } if priority > maxPriority { maxPriority = priority