1 files changed, 7 insertions, 1 deletions

diff --git a/include/linux/sched.h b/include/linux/sched.h
index cbbfca69aa4a..c472414953bf 100644
--- a/include/linux/sched.h
+++ b/include/linux/sched.h
@@ -2293,8 +2293,14 @@ extern int _cond_resched(void);
 
 extern int __cond_resched_lock(spinlock_t *lock);
 
+#ifdef CONFIG_PREEMPT
+#define PREEMPT_LOCK_OFFSET     PREEMPT_OFFSET
+#else
+#define PREEMPT_LOCK_OFFSET     0
+#endif
+
 #define cond_resched_lock(lock) ({                              \
-        __might_sleep(__FILE__, __LINE__, PREEMPT_OFFSET);      \
+        __might_sleep(__FILE__, __LINE__, PREEMPT_LOCK_OFFSET); \
         __cond_resched_lock(lock);                              \
 })