struct vm_area_struct *vma = walk->vma;
        bool locked = !!(vma->vm_flags & VM_LOCKED);
        struct page *page = NULL;
+       struct folio *folio;
        bool migration = false;
 
        if (pmd_present(*pmd)) {
        }
        if (IS_ERR_OR_NULL(page))
                return;
-       if (PageAnon(page))
+       folio = page_folio(page);
+       if (folio_test_anon(folio))
                mss->anonymous_thp += HPAGE_PMD_SIZE;
-       else if (PageSwapBacked(page))
+       else if (folio_test_swapbacked(folio))
                mss->shmem_thp += HPAGE_PMD_SIZE;
-       else if (is_zone_device_page(page))
+       else if (folio_is_zone_device(folio))
                /* pass */;
        else
                mss->file_thp += HPAGE_PMD_SIZE;