[PATCH v2 2/2] powerpc/mm/hugetlb: Switch hugetlb update to use pte_update

["Aneesh Kumar K.V" <aneesh.kumar@linux.vnet.ibm.com>]

Now that we have updated hugetlb functions to take vm_area_struct and we can
derive huge page size from vma, switch the pte update to use generic functions.

Signed-off-by: Aneesh Kumar K.V <aneesh.kumar@linux.vnet.ibm.com>
---
 arch/powerpc/include/asm/book3s/64/hugetlb.h | 34 +++++++---------------------
 arch/powerpc/include/asm/hugetlb.h           |  2 +-
 2 files changed, 9 insertions(+), 27 deletions(-)

diff --git a/arch/powerpc/include/asm/book3s/64/hugetlb.h b/arch/powerpc/include/asm/book3s/64/hugetlb.h
index 80fa0c828413..0a6db2086140 100644
--- a/arch/powerpc/include/asm/book3s/64/hugetlb.h
+++ b/arch/powerpc/include/asm/book3s/64/hugetlb.h
@@ -31,36 +31,18 @@ static inline int hstate_get_psize(struct hstate *hstate)
 	}
 }
 
-static inline unsigned long huge_pte_update(struct mm_struct *mm, unsigned long addr,
+static inline unsigned long huge_pte_update(struct vm_area_struct *vma, unsigned long addr,
 					    pte_t *ptep, unsigned long clr,
 					    unsigned long set)
 {
-	if (radix_enabled()) {
-		unsigned long old_pte;
+	unsigned long pg_sz;
 
-		if (cpu_has_feature(CPU_FTR_POWER9_DD1)) {
+	VM_WARN_ON(!is_vm_hugetlb_page(vma));
+	pg_sz = huge_page_size(hstate_vma(vma));
 
-			unsigned long new_pte;
-
-			old_pte = __radix_pte_update(ptep, ~0, 0);
-			asm volatile("ptesync" : : : "memory");
-			/*
-			 * new value of pte
-			 */
-			new_pte = (old_pte | set) & ~clr;
-			/*
-			 * For now let's do heavy pid flush
-			 * radix__flush_tlb_page_psize(mm, addr, mmu_virtual_psize);
-			 */
-			radix__flush_tlb_mm(mm);
-
-			__radix_pte_update(ptep, 0, new_pte);
-		} else
-			old_pte = __radix_pte_update(ptep, clr, set);
-		asm volatile("ptesync" : : : "memory");
-		return old_pte;
-	}
-	return hash__pte_update(mm, addr, ptep, clr, set, true);
+	if (radix_enabled())
+		return radix__pte_update(vma->vm_mm, addr, ptep, clr, set, pg_sz);
+	return hash__pte_update(vma->vm_mm, addr, ptep, clr, set, true);
 }
 
 static inline void huge_ptep_set_wrprotect(struct vm_area_struct *vma,
@@ -69,7 +51,7 @@ static inline void huge_ptep_set_wrprotect(struct vm_area_struct *vma,
 	if ((pte_raw(*ptep) & cpu_to_be64(_PAGE_WRITE)) == 0)
 		return;
 
-	huge_pte_update(vma->vm_mm, addr, ptep, _PAGE_WRITE, 0);
+	huge_pte_update(vma, addr, ptep, _PAGE_WRITE, 0);
 }
 
 #endif
diff --git a/arch/powerpc/include/asm/hugetlb.h b/arch/powerpc/include/asm/hugetlb.h
index bb1bf23d6f90..f0731dff76c2 100644
--- a/arch/powerpc/include/asm/hugetlb.h
+++ b/arch/powerpc/include/asm/hugetlb.h
@@ -136,7 +136,7 @@ static inline pte_t huge_ptep_get_and_clear(struct vm_area_struct *vma,
 					    unsigned long addr, pte_t *ptep)
 {
 #ifdef CONFIG_PPC64
-	return __pte(huge_pte_update(vma->vm_mm, addr, ptep, ~0UL, 0));
+	return __pte(huge_pte_update(vma, addr, ptep, ~0UL, 0));
 #else
 	return __pte(pte_update(ptep, ~0UL, 0));
 #endif
-- 
2.10.2