1 files changed, 5 insertions, 1 deletions

diff --git a/kernel/softirq.c b/kernel/softirq.c
index de2f57fddc04..900dcfee542c 100644
--- a/kernel/softirq.c
+++ b/kernel/softirq.c
@@ -139,9 +139,13 @@ static void __local_bh_enable(unsigned int cnt)
 {
         lockdep_assert_irqs_disabled();
 
+        if (preempt_count() == cnt)
+                trace_preempt_on(CALLER_ADDR0, get_lock_parent_ip());
+
         if (softirq_count() == (cnt & SOFTIRQ_MASK))
                 trace_softirqs_on(_RET_IP_);
-        preempt_count_sub(cnt);
+
+        __preempt_count_sub(cnt);
 }
 
 /*