if (pte_cont(*ptep)) {
                int ncontig, i;
                size_t pgsize;
-               pte_t *cpte;
                bool is_dirty = false;
 
-               cpte = huge_pte_offset(mm, addr);
-               ncontig = find_num_contig(mm, addr, cpte, &pgsize);
+               ncontig = find_num_contig(mm, addr, ptep, &pgsize);
                /* save the 1st pte to return */
-               pte = ptep_get_and_clear(mm, addr, cpte);
+               pte = ptep_get_and_clear(mm, addr, ptep);
                for (i = 1, addr += pgsize; i < ncontig; ++i, addr += pgsize) {
                        /*
                         * If HW_AFDBM is enabled, then the HW could
                         * turn on the dirty bit for any of the page
                         * in the set, so check them all.
                         */
-                       ++cpte;
-                       if (pte_dirty(ptep_get_and_clear(mm, addr, cpte)))
+                       ++ptep;
+                       if (pte_dirty(ptep_get_and_clear(mm, addr, ptep)))
                                is_dirty = true;
                }
                if (is_dirty)
                               unsigned long addr, pte_t *ptep,
                               pte_t pte, int dirty)
 {
-       pte_t *cpte;
-
        if (pte_cont(pte)) {
                int ncontig, i, changed = 0;
                size_t pgsize = 0;
                        __pgprot(pte_val(pfn_pte(pfn, __pgprot(0))) ^
                                 pte_val(pte));
 
-               cpte = huge_pte_offset(vma->vm_mm, addr);
-               pfn = pte_pfn(*cpte);
-               ncontig = find_num_contig(vma->vm_mm, addr, cpte,
+               pfn = pte_pfn(pte);
+               ncontig = find_num_contig(vma->vm_mm, addr, ptep,
                                          &pgsize);
-               for (i = 0; i < ncontig; ++i, ++cpte, addr += pgsize) {
-                       changed |= ptep_set_access_flags(vma, addr, cpte,
+               for (i = 0; i < ncontig; ++i, ++ptep, addr += pgsize) {
+                       changed |= ptep_set_access_flags(vma, addr, ptep,
                                                        pfn_pte(pfn,
                                                                hugeprot),
                                                        dirty);
 {
        if (pte_cont(*ptep)) {
                int ncontig, i;
-               pte_t *cpte;
                size_t pgsize = 0;
 
-               cpte = huge_pte_offset(mm, addr);
-               ncontig = find_num_contig(mm, addr, cpte, &pgsize);
-               for (i = 0; i < ncontig; ++i, ++cpte, addr += pgsize)
-                       ptep_set_wrprotect(mm, addr, cpte);
+               ncontig = find_num_contig(mm, addr, ptep, &pgsize);
+               for (i = 0; i < ncontig; ++i, ++ptep, addr += pgsize)
+                       ptep_set_wrprotect(mm, addr, ptep);
        } else {
                ptep_set_wrprotect(mm, addr, ptep);
        }
 {
        if (pte_cont(*ptep)) {
                int ncontig, i;
-               pte_t *cpte;
                size_t pgsize = 0;
 
-               cpte = huge_pte_offset(vma->vm_mm, addr);
-               ncontig = find_num_contig(vma->vm_mm, addr, cpte,
+               ncontig = find_num_contig(vma->vm_mm, addr, ptep,
                                          &pgsize);
-               for (i = 0; i < ncontig; ++i, ++cpte, addr += pgsize)
-                       ptep_clear_flush(vma, addr, cpte);
+               for (i = 0; i < ncontig; ++i, ++ptep, addr += pgsize)
+                       ptep_clear_flush(vma, addr, ptep);
        } else {
                ptep_clear_flush(vma, addr, ptep);
        }