From 8f77a39b50d164f75866edb720e10704e608d916 Mon Sep 17 00:00:00 2001
From: Nadav Har'El <nyh@math.technion.ac.il>
Date: Mon, 27 Feb 2012 15:07:29 +0200
Subject: [PATCH] --- yaml --- r: 296245 b: refs/heads/master c:
 d550dda192c1bd039afb774b99485e88b70d7cb8 h: refs/heads/master i:   296243:
 7e4ec318d30611b43722fb99b3c5ea83d636b5c9 v: v3

---
 [refs]                      | 2 +-
 trunk/drivers/vhost/vhost.c | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/[refs] b/[refs]
index 3947c7f5cd97..ff6dc0c73a26 100644
--- a/[refs]
+++ b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: b17d5c6e190f3d328aae0444f8b93d58d0015714
+refs/heads/master: d550dda192c1bd039afb774b99485e88b70d7cb8
diff --git a/trunk/drivers/vhost/vhost.c b/trunk/drivers/vhost/vhost.c
index c14c42b95ab8..ae66278e4dcf 100644
--- a/trunk/drivers/vhost/vhost.c
+++ b/trunk/drivers/vhost/vhost.c
@@ -222,6 +222,8 @@ static int vhost_worker(void *data)
 		if (work) {
 			__set_current_state(TASK_RUNNING);
 			work->fn(work);
+			if (need_resched())
+				schedule();
 		} else
 			schedule();