[PATCH 1/1] arm64: preempt: Add NEED_RESCHED to the preempt_count()

[wonder_rock at 126.com]

From: Deyan Wang <wonder_rock at 126.com>

preempt_count() will be checked in preempt_schedule_irq(),
we need check if NEED_RESCHED in preempt point.

Signed-off-by: Deyan Wang <wonder_rock at 126.com>
---
 arch/arm64/include/asm/preempt.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/arch/arm64/include/asm/preempt.h b/arch/arm64/include/asm/preempt.h
index 0159b625cc7f..083b0fd3c52a 100644
--- a/arch/arm64/include/asm/preempt.h
+++ b/arch/arm64/include/asm/preempt.h
@@ -10,7 +10,7 @@
 
 static inline int preempt_count(void)
 {
-	return READ_ONCE(current_thread_info()->preempt.count);
+	return READ_ONCE(current_thread_info()->preempt_count);
 }
 
 static inline void preempt_count_set(u64 pc)
-- 
2.25.1