struct dev_pagemap **pgmap)
 {
        struct mm_struct *mm = vma->vm_mm;
+       struct folio *folio;
        struct page *page;
        spinlock_t *ptl;
        pte_t *ptep, pte;
                        goto out;
                }
        }
+       folio = page_folio(page);
 
        if (!pte_write(pte) && gup_must_unshare(vma, flags, page)) {
                page = ERR_PTR(-EMLINK);
                       !PageAnonExclusive(page), page);
 
        /* try_grab_folio() does nothing unless FOLL_GET or FOLL_PIN is set. */
-       ret = try_grab_folio(page_folio(page), 1, flags);
+       ret = try_grab_folio(folio, 1, flags);
        if (unlikely(ret)) {
                page = ERR_PTR(ret);
                goto out;
         * Documentation/core-api/pin_user_pages.rst for details.
         */
        if (flags & FOLL_PIN) {
-               ret = arch_make_page_accessible(page);
+               ret = arch_make_folio_accessible(folio);
                if (ret) {
                        unpin_user_page(page);
                        page = ERR_PTR(ret);
                 * details.
                 */
                if (flags & FOLL_PIN) {
-                       ret = arch_make_page_accessible(page);
+                       ret = arch_make_folio_accessible(folio);
                        if (ret) {
                                gup_put_folio(folio, 1, flags);
                                goto pte_unmap;