1 files changed, 1 insertions, 6 deletions

diff --git a/arch/arm/mach-tegra/pm.c b/arch/arm/mach-tegra/pm.c
index 0494f739c95f..acacbe8d1afc 100644
--- a/arch/arm/mach-tegra/pm.c
+++ b/arch/arm/mach-tegra/pm.c
@@ -164,12 +164,7 @@ bool tegra_set_cpu_in_lp2(int phy_cpu_id)
 
 static int tegra_sleep_cpu(unsigned long v2p)
 {
-        /* Switch to the identity mapping. */
-        cpu_switch_mm(idmap_pgd, &init_mm);
-
-        /* Flush the TLB. */
-        local_flush_tlb_all();
-
+        setup_mm_for_reboot();
         tegra_sleep_cpu_finish(v2p);
 
         /* should never here */