put_disk(pmem->disk);
 }
 
-static void pmem_pagemap_page_free(struct page *page, void *data)
+static void pmem_pagemap_page_free(struct page *page)
 {
        wake_up_var(&page->_refcount);
 }
 
         * reach 0 refcount unless there is a refcount bug. This allows the
         * device driver to implement its own memory management.)
         */
-       void (*page_free)(struct page *page, void *data);
+       void (*page_free)(struct page *page);
 
        /*
         * Transition the refcount in struct dev_pagemap to the dead state.
        struct resource res;
        struct percpu_ref *ref;
        struct device *dev;
-       void *data;
        enum memory_type type;
        u64 pci_p2pdma_bus_offset;
        const struct dev_pagemap_ops *ops;
 
 
                mem_cgroup_uncharge(page);
 
-               page->pgmap->ops->page_free(page, page->pgmap->data);
+               page->pgmap->ops->page_free(page);
        } else if (!count)
                __put_page(page);
 }
 
 
 static vm_fault_t hmm_devmem_migrate_to_ram(struct vm_fault *vmf)
 {
-       struct hmm_devmem *devmem = vmf->page->pgmap->data;
+       struct hmm_devmem *devmem =
+               container_of(vmf->page->pgmap, struct hmm_devmem, pagemap);
 
        return devmem->ops->fault(devmem, vmf->vma, vmf->address, vmf->page,
                        vmf->flags, vmf->pmd);
 }
 
-static void hmm_devmem_free(struct page *page, void *data)
+static void hmm_devmem_free(struct page *page)
 {
-       struct hmm_devmem *devmem = data;
+       struct hmm_devmem *devmem =
+               container_of(page->pgmap, struct hmm_devmem, pagemap);
 
        devmem->ops->free(devmem, page);
 }
        devmem->pagemap.ops = &hmm_pagemap_ops;
        devmem->pagemap.altmap_valid = false;
        devmem->pagemap.ref = &devmem->ref;
-       devmem->pagemap.data = devmem;
 
        result = devm_memremap_pages(devmem->device, &devmem->pagemap);
        if (IS_ERR(result))