2 files changed, 14 insertions, 0 deletions

diff --git a/arch/x86/mm/init_32.c b/arch/x86/mm/init_32.c
index 0a3707fb973b..3f91bdc20971 100644
--- a/arch/x86/mm/init_32.c
+++ b/arch/x86/mm/init_32.c
@@ -1054,10 +1054,16 @@ unsigned long __init_refok init_memory_mapping(unsigned long start,
                                                    mr[i].page_size_mask);
 #endif
 
+#ifdef CONFIG_X86_32
         early_ioremap_page_table_range_init();
 
         load_cr3(swapper_pg_dir);
+#endif
 
+#ifdef CONFIG_X86_64
+        if (!after_bootmem)
+                mmu_cr4_features = read_cr4();
+#endif
         __flush_tlb_all();
 
         if (!after_bootmem && table_end > table_start)
diff --git a/arch/x86/mm/init_64.c b/arch/x86/mm/init_64.c
index e4fadea2e521..5ecb23a57d2f 100644
--- a/arch/x86/mm/init_64.c
+++ b/arch/x86/mm/init_64.c
@@ -832,8 +832,16 @@ unsigned long __init_refok init_memory_mapping(unsigned long start,
                                                    mr[i].page_size_mask);
 #endif
 
+#ifdef CONFIG_X86_32
+        early_ioremap_page_table_range_init();
+
+        load_cr3(swapper_pg_dir);
+#endif
+
+#ifdef CONFIG_X86_64
         if (!after_bootmem)
                 mmu_cr4_features = read_cr4();
+#endif
         __flush_tlb_all();
 
         if (!after_bootmem && table_end > table_start)