X-Git-Url: https://git.zerfleddert.de/cgi-bin/gitweb.cgi/ms2-kexec/blobdiff_plain/4e93cb000786b18ffd9929f33055cad2a7cebbcf..HEAD:/mmu.c

diff --git a/mmu.c b/mmu.c
index 6d4bce3..f58a7bd 100644
--- 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();
 }