#define VMALLOC_SIZE_TB_L4     32UL
 #define VMALLOC_SIZE_TB_L5     12800UL
 
-#ifdef CONFIG_X86_5LEVEL
-# define __VMEMMAP_BASE                _AC(0xffd4000000000000, UL)
-#else
-# define __VMEMMAP_BASE                _AC(0xffffea0000000000, UL)
-#endif
+#define __VMEMMAP_BASE_L4      0xffffea0000000000
+#define __VMEMMAP_BASE_L5      0xffd4000000000000
 
 #ifdef CONFIG_DYNAMIC_MEMORY_LAYOUT
 # define VMALLOC_START         vmalloc_base
 #else
 # define VMALLOC_START         __VMALLOC_BASE_L4
 # define VMALLOC_SIZE_TB       VMALLOC_SIZE_TB_L4
-# define VMEMMAP_START         __VMEMMAP_BASE
+# define VMEMMAP_START         __VMEMMAP_BASE_L4
 #endif /* CONFIG_DYNAMIC_MEMORY_LAYOUT */
 
 #define VMALLOC_END            (VMALLOC_START + (VMALLOC_SIZE_TB << 40) - 1)
 
 EXPORT_SYMBOL(page_offset_base);
 unsigned long vmalloc_base __ro_after_init = __VMALLOC_BASE_L4;
 EXPORT_SYMBOL(vmalloc_base);
-unsigned long vmemmap_base __ro_after_init = __VMEMMAP_BASE;
+unsigned long vmemmap_base __ro_after_init = __VMEMMAP_BASE_L4;
 EXPORT_SYMBOL(vmemmap_base);
 #endif
 
        *fixup_int(&ptrs_per_p4d, physaddr) = 512;
        *fixup_long(&page_offset_base, physaddr) = __PAGE_OFFSET_BASE_L5;
        *fixup_long(&vmalloc_base, physaddr) = __VMALLOC_BASE_L5;
+       *fixup_long(&vmemmap_base, physaddr) = __VMEMMAP_BASE_L5;
 }
 #else
 static void __head check_la57_support(unsigned long physaddr) {}