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;
                 }