update offsets to new european GB kernel, set software booting from USB

[ms2-kexec] / mmu.c

diff --git a/mmu.c b/mmu.c
index 6d4bce35af99f4f4da20b1c9be813796cf6c2dc6..f58a7bdb01b0bf20ffb2818d3fb73b67a8e0c80c 100644 (file)
--- a/mmu.c
+++ b/mmu.c
@@ -16,8 +16,8 @@
 #include <linux/mman.h>
 #include <linux/nodemask.h>
 #include <linux/ioport.h>
-#include <linux/lttlite-events.h>
 
+#include "tlbflush.h"
 #include <asm/cputype.h>
 #include <asm/mach-types.h>
 #include <asm/sections.h>
@@ -42,10 +42,7 @@ void setup_mm_for_reboot(char mode)
        pgd_t *pgd;
        int i;
 
-       if (current->mm && current->mm->pgd)
-               pgd = current->mm->pgd;
-       else
-               pgd = init_mm.pgd;
+       pgd = current->mm->pgd;
 
        base_pmdval = PMD_SECT_AP_WRITE | PMD_SECT_AP_READ | PMD_TYPE_SECT;
 
@@ -58,4 +55,5 @@ void setup_mm_for_reboot(char mode)
                pmd[1] = __pmd(pmdval + (1 << (PGDIR_SHIFT - 1)));
                flush_pmd_entry(pmd);
        }
+       my_local_flush_tlb_all();
 }