2 files changed, 12 insertions, 3 deletions

diff --git a/arch/xtensa/kernel/head.S b/arch/xtensa/kernel/head.S
index ef12c0e6fa25..7d740ebbe198 100644
--- a/arch/xtensa/kernel/head.S
+++ b/arch/xtensa/kernel/head.S
@@ -68,6 +68,15 @@ _SetupMMU:
 
 #ifdef CONFIG_INITIALIZE_XTENSA_MMU_INSIDE_VMLINUX
         initialize_mmu
+#if defined(CONFIG_MMU) && XCHAL_HAVE_PTP_MMU && XCHAL_HAVE_SPANNING_WAY
+        rsr     a2, excsave1
+        movi    a3, 0x08000000
+        bgeu    a2, a3, 1f
+        movi    a3, 0xd0000000
+        add     a2, a2, a3
+        wsr     a2, excsave1
+1:
+#endif
 #endif
         .end    no-absolute-literals
 
diff --git a/arch/xtensa/kernel/setup.c b/arch/xtensa/kernel/setup.c
index 6dd25ecde3f5..14c6c3a6f04b 100644
--- a/arch/xtensa/kernel/setup.c
+++ b/arch/xtensa/kernel/setup.c
@@ -152,8 +152,8 @@ static int __init parse_tag_initrd(const bp_tag_t* tag)
 {
         meminfo_t* mi;
         mi = (meminfo_t*)(tag->data);
-        initrd_start = (void*)(mi->start);
-        initrd_end = (void*)(mi->end);
+        initrd_start = __va(mi->start);
+        initrd_end = __va(mi->end);
 
         return 0;
 }
@@ -164,7 +164,7 @@ __tagtable(BP_TAG_INITRD, parse_tag_initrd);
 
 static int __init parse_tag_fdt(const bp_tag_t *tag)
 {
-        dtb_start = (void *)(tag->data[0]);
+        dtb_start = __va(tag->data[0]);
         return 0;
 }