* not much we can do. We just print a message and ignore otherwise.
  */
 
+#define FSDAX_INVALID_PGOFF ULONG_MAX
+
 /*
  * Schedule a process for later kill.
  * Uses GFP_ATOMIC allocations to avoid potential recursions in the VM.
  *
- * Notice: @fsdax_pgoff is used only when @p is a fsdax page.
- *   In other cases, such as anonymous and file-backend page, the address to be
- *   killed can be caculated by @p itself.
+ * Note: @fsdax_pgoff is used only when @p is a fsdax page and a
+ * filesystem with a memory failure handler has claimed the
+ * memory_failure event. In all other cases, page->index and
+ * page->mapping are sufficient for mapping the page back to its
+ * corresponding user virtual address.
  */
 static void add_to_kill(struct task_struct *tsk, struct page *p,
                        pgoff_t fsdax_pgoff, struct vm_area_struct *vma,
 
        tk->addr = page_address_in_vma(p, vma);
        if (is_zone_device_page(p)) {
-               /*
-                * Since page->mapping is not used for fsdax, we need
-                * calculate the address based on the vma.
-                */
-               if (p->pgmap->type == MEMORY_DEVICE_FS_DAX)
+               if (fsdax_pgoff != FSDAX_INVALID_PGOFF)
                        tk->addr = vma_pgoff_address(fsdax_pgoff, 1, vma);
                tk->size_shift = dev_pagemap_mapping_shift(vma, tk->addr);
        } else
                        if (!page_mapped_in_vma(page, vma))
                                continue;
                        if (vma->vm_mm == t->mm)
-                               add_to_kill(t, page, 0, vma, to_kill);
+                               add_to_kill(t, page, FSDAX_INVALID_PGOFF, vma,
+                                           to_kill);
                }
        }
        read_unlock(&tasklist_lock);
                         * to be informed of all such data corruptions.
                         */
                        if (vma->vm_mm == t->mm)
-                               add_to_kill(t, page, 0, vma, to_kill);
+                               add_to_kill(t, page, FSDAX_INVALID_PGOFF, vma,
+                                           to_kill);
                }
        }
        read_unlock(&tasklist_lock);