tools/testing/selftests/vm/gup_benchmark.c: add MAP_HUGETLB option

Add a new option '-H' to the gup benchmark to help understand how hugetlb
mapping pages compare with the default.

Link: http://lkml.kernel.org/r/20181010195605.10689-6-keith.busch@intel.com
Signed-off-by: Keith Busch <keith.busch@intel.com>
Reviewed-by: Andrew Morton <akpm@linux-foundation.org>
Cc: Kirill Shutemov <kirill.shutemov@linux.intel.com>
Cc: Dave Hansen <dave.hansen@intel.com>
Cc: Dan Williams <dan.j.williams@intel.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>

1 files changed, 4 insertions, 1 deletions

diff --git a/tools/testing/selftests/vm/gup_benchmark.c b/tools/testing/selftests/vm/gup_benchmark.c
index 5c8e4cb1441a..880b96fc80d4 100644
--- a/tools/testing/selftests/vm/gup_benchmark.c
+++ b/tools/testing/selftests/vm/gup_benchmark.c
@@ -36,7 +36,7 @@ int main(int argc, char **argv)
         char *file = "/dev/zero";
         char *p;
 
-        while ((opt = getopt(argc, argv, "m:r:n:f:tTLUS")) != -1) {
+        while ((opt = getopt(argc, argv, "m:r:n:f:tTLUSH")) != -1) {
                 switch (opt) {
                 case 'm':
                         size = atoi(optarg) * MB;
@@ -69,6 +69,9 @@ int main(int argc, char **argv)
                         flags &= ~MAP_PRIVATE;
                         flags |= MAP_SHARED;
                         break;
+                case 'H':
+                        flags |= MAP_HUGETLB;
+                        break;
                 default:
                         return -1;
                 }