[PATCH v2] /dev/zero: try to align PMD_SIZE for private mapping

[Zhang Qilong]

If THP is enable, mapping /dev/zero with a size larger than 2MB could
achieve performance gains by allocating aligned address. The mprot_tw4m
in libMicro average execution time on arm64:
  - Test case:        mprot_tw4m
  - Before the patch:   22 us
  - After the patch:    17 us

Signed-off-by: Zhang Qilong <zhangqilong3@xxxxxxxxxx>
---
 drivers/char/mem.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/drivers/char/mem.c b/drivers/char/mem.c
index 48839958b0b1..3699e716b84e 100644
--- a/drivers/char/mem.c
+++ b/drivers/char/mem.c
@@ -525,12 +525,20 @@ static unsigned long get_unmapped_area_zero(struct file *file,
 		 * so as not to confuse shmem with our handle on "/dev/zero".
 		 */
 		return shmem_get_unmapped_area(NULL, addr, len, pgoff, flags);
 	}
 
+#ifdef CONFIG_TRANSPARENT_HUGEPAGE
+	/*
+	 * Attempt to map aligned to huge page size if possible, otherwise we
+	 * fall back to system page size mappings in thp_get_unmapped_area.
+	 */
+	return thp_get_unmapped_area(file, addr, len, pgoff, flags);
+#else
 	/* Otherwise flags & MAP_PRIVATE: with no shmem object beneath it */
 	return mm_get_unmapped_area(current->mm, file, addr, len, pgoff, flags);
+#endif
 #else
 	return -ENOSYS;
 #endif
 }
 
-- 
2.43.0