1 files changed, 2 insertions, 0 deletions

diff --git a/arch/mips/mm/init.c b/arch/mips/mm/init.c
index faa5c9822ecc..198a3147dd7d 100644
--- a/arch/mips/mm/init.c
+++ b/arch/mips/mm/init.c
@@ -90,6 +90,7 @@ static void *__kmap_pgprot(struct page *page, unsigned long addr, pgprot_t prot)
 
         BUG_ON(Page_dcache_dirty(page));
 
+        preempt_disable();
         pagefault_disable();
         idx = (addr >> PAGE_SHIFT) & (FIX_N_COLOURS - 1);
         idx += in_interrupt() ? FIX_N_COLOURS : 0;
@@ -152,6 +153,7 @@ void kunmap_coherent(void)
         write_c0_entryhi(old_ctx);
         local_irq_restore(flags);
         pagefault_enable();
+        preempt_enable();
 }
 
 void copy_user_highpage(struct page *to, struct page *from,