signed-off-by: Andrew Theurer <redacted>

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index fbf1fd0..c767915 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c

@@ -4844,6 +4844,9 @@ bool __sched yield_to(struct task_struct *p, bool

preempt)

 again:
 	p_rq = task_rq(p);
+	if (task_running(p_rq, p) || p->state || !(p_rq->curr->flags &
PF_VCPU)) {
+		goto out_no_unlock;
+	}
 	double_rq_lock(rq, p_rq);
 	while (task_rq(p) != p_rq) {
 		double_rq_unlock(rq, p_rq);

@@ -4856,8 +4859,6 @@ again:
 	if (curr->sched_class != p->sched_class)
 		goto out;

-	if (task_running(p_rq, p) || p->state)
-		goto out;

 	yielded = curr->sched_class->yield_to_task(rq, p, preempt);
 	if (yielded) {

@@ -4879,6 +4880,7 @@ again:

 out:
 	double_rq_unlock(rq, p_rq);
+out_no_unlock:
 	local_irq_restore(flags);

 	if (yielded)