powerpc/mm: Add SMP support to no-hash TLB handling

[deliverable/linux.git] / arch / powerpc / include / asm / highmem.h

diff --git a/arch/powerpc/include/asm/highmem.h b/arch/powerpc/include/asm/highmem.h
index 5d99b6489d569d44e54a3781734e797d72100b5c..fd97e501aa6a6a8c865cd3f89ea2abc21fd3ec6f 100644 (file)
--- a/arch/powerpc/include/asm/highmem.h
+++ b/arch/powerpc/include/asm/highmem.h
@@ -84,8 +84,8 @@ static inline void *kmap_atomic_prot(struct page *page, enum km_type type, pgpro
 #ifdef CONFIG_DEBUG_HIGHMEM
        BUG_ON(!pte_none(*(kmap_pte-idx)));
 #endif
-       set_pte_at(&init_mm, vaddr, kmap_pte-idx, mk_pte(page, prot));
-       flush_tlb_page(NULL, vaddr);
+       __set_pte_at(&init_mm, vaddr, kmap_pte-idx, mk_pte(page, prot));
+       local_flush_tlb_page(NULL, vaddr);
 
        return (void*) vaddr;
 }
@@ -113,7 +113,7 @@ static inline void kunmap_atomic(void *kvaddr, enum km_type type)
         * this pte without first remap it
         */
        pte_clear(&init_mm, vaddr, kmap_pte-idx);
-       flush_tlb_page(NULL, vaddr);
+       local_flush_tlb_page(NULL, vaddr);
 #endif
        pagefault_enable();
 }