1 files changed, 3 insertions, 3 deletions

diff --git a/arch/powerpc/kernel/time.c b/arch/powerpc/kernel/time.c
index fe6f3a285455..a32823dcd9a4 100644
--- a/arch/powerpc/kernel/time.c
+++ b/arch/powerpc/kernel/time.c
@@ -244,7 +244,7 @@ static u64 scan_dispatch_log(u64 stop_tb)
 void accumulate_stolen_time(void)
 {
         u64 sst, ust;
-        u8 save_soft_enabled = local_paca->soft_enabled;
+        unsigned long save_irq_soft_mask = irq_soft_mask_return();
         struct cpu_accounting_data *acct = &local_paca->accounting;
 
         /* We are called early in the exception entry, before
@@ -253,7 +253,7 @@ void accumulate_stolen_time(void)
          * needs to reflect that so various debug stuff doesn't
          * complain
          */
-        local_paca->soft_enabled = 0;
+        irq_soft_mask_set(IRQS_DISABLED);
 
         sst = scan_dispatch_log(acct->starttime_user);
         ust = scan_dispatch_log(acct->starttime);
@@ -261,7 +261,7 @@ void accumulate_stolen_time(void)
         acct->utime -= ust;
         acct->steal_time += ust + sst;
 
-        local_paca->soft_enabled = save_soft_enabled;
+        irq_soft_mask_set(save_irq_soft_mask);
 }
 
 static inline u64 calculate_stolen_time(u64 stop_tb)