__collapse_huge_page_isolate, fix with scan pmd

static int __collapse_huge_page_isolate(struct vm_area_struct *vma,
                                        unsigned long address,
                                        pte_t *pte)
{
        struct page *page;
        pte_t *_pte;
        int referenced = 0, none = 0;
        bool read_only_pte = false;
        for (_pte = pte; _pte < pte+HPAGE_PMD_NR;
             _pte++, address += PAGE_SIZE) {
                pte_t pteval = *_pte;

                if (pte_none(pteval)) {
                        if (++none <= khugepaged_max_ptes_none)
                                continue;
                        else
                                goto out;
                }
                if (!pte_present(pteval))
                        pr_info("is not present, pid: %i\n", current->pid);
                        goto out;
                if (!pte_write(pteval)) {
                        pr_info("read-only pte\n");
                        read_only_pte = true;
                }
                page = vm_normal_page(vma, address, pteval);
                if (unlikely(!page))
                        pr_info("!page\n");
                        goto out;

                if (read_only_pte) {
                        read_only_pte = false;
                        if (PageSwapCache(page))
                                reuse_swap_page(page);
                }
                pr_info("vm_bug_on\n");
                VM_BUG_ON_PAGE(PageCompound(page), page);
                VM_BUG_ON_PAGE(!PageAnon(page), page);
                VM_BUG_ON_PAGE(!PageSwapBacked(page), page);
                /* cannot use mapcount: can't collapse if there's a gup pin */

                if (page_count(page) != 1 + !!PageSwapCache(page))
                        pr_info("page_count() error\n");
                        goto out;
                /* pr_info("after page_count\n"); */
                /*
                 * We can do it before isolate_lru_page because the
                 * page can't be freed from under us. NOTE: PG_lock
                 * is needed to serialize against split_huge_page
                 * when invoked from the VM.
                 */
                if (!trylock_page(page))
                        pr_info("trylock_page() error\n");
                        goto out;
                /*
                 * Isolate the page to avoid collapsing an hugepage
                 * currently in use by the VM.
                 */
                if (isolate_lru_page(page)) {
                        pr_info("isolate_lru_page() error\n");
                        unlock_page(page);
                        goto out;
                }
                /* 0 stands for page_is_file_cache(page) == false */
                inc_zone_page_state(page, NR_ISOLATED_ANON + 0);
                VM_BUG_ON_PAGE(!PageLocked(page), page);
                VM_BUG_ON_PAGE(PageLRU(page), page);
                pr_info("page on page lru succeeds\n");

                /* If there is no mapped pte young don't collapse the page */
                if (pte_young(pteval) || PageReferenced(page) ||
                    mmu_notifier_test_young(vma->vm_mm, address)) {
                        referenced = 1;
                }
        }
        if (likely(referenced))
                pr_info("referenced set 1\n");
                return 1;
out:
        release_pte_pages(pte, _pte);
        pr_info("ptes released\n");
        return 0;
}