1 files changed, 3 insertions, 2 deletions

diff --git a/arch/ppc/mm/init.c b/arch/ppc/mm/init.c
index 33ada72c7330..32ee49710267 100644
--- a/arch/ppc/mm/init.c
+++ b/arch/ppc/mm/init.c
@@ -560,8 +560,9 @@ void flush_dcache_page(struct page *page)
 void flush_dcache_icache_page(struct page *page)
 {
 #ifdef CONFIG_BOOKE
-        __flush_dcache_icache(kmap(page));
-        kunmap(page);
+        void *start = kmap_atomic(page, KM_PPC_SYNC_ICACHE);
+        __flush_dcache_icache(start);
+        kunmap_atomic(start, KM_PPC_SYNC_ICACHE);
 #elif CONFIG_8xx
         /* On 8xx there is no need to kmap since highmem is not supported */
         __flush_dcache_icache(page_address(page));