1 files changed, 1 insertions, 1 deletions

diff --git a/arch/x86/include/asm/mmu_context.h b/arch/x86/include/asm/mmu_context.h
index ecfcb6643c9b..265c907d7d4c 100644
--- a/arch/x86/include/asm/mmu_context.h
+++ b/arch/x86/include/asm/mmu_context.h
@@ -293,7 +293,7 @@ static inline unsigned long __get_current_cr3_fast(void)
         unsigned long cr3 = __pa(this_cpu_read(cpu_tlbstate.loaded_mm)->pgd);
 
         /* For now, be very restrictive about when this can be called. */
-        VM_WARN_ON(in_nmi() || !in_atomic());
+        VM_WARN_ON(in_nmi() || preemptible());
 
         VM_BUG_ON(cr3 != __read_cr3());
         return cr3;