From e410c0fd0057e018af1e67d878707322192428d3 Mon Sep 17 00:00:00 2001
From: Jacob Khaliqi <jacobkhaliqi@gmail.com>
Date: Mon, 10 Jun 2024 18:46:41 -0700
Subject: [PATCH] Add support for parquet_prefetch_rowgroups flag in
 TpchBenchmark (#9873)

Summary:
Resolves https://github.com/facebookincubator/velox/issues/8693

Pull Request resolved: https://github.com/facebookincubator/velox/pull/9873

Reviewed By: bikramSingh91

Differential Revision: D58370045

Pulled By: kevinwilfong

fbshipit-source-id: a8ce5f1534c53c3e797c72f784a59feaba6030de
---
 velox/benchmarks/tpch/TpchBenchmark.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/velox/benchmarks/tpch/TpchBenchmark.cpp b/velox/benchmarks/tpch/TpchBenchmark.cpp
index 2405714ea906..fcf8c5c9a9d7 100644
--- a/velox/benchmarks/tpch/TpchBenchmark.cpp
+++ b/velox/benchmarks/tpch/TpchBenchmark.cpp
@@ -263,6 +263,8 @@ class TpchBenchmark {
     configurationValues
         [connector::hive::HiveConfig::kMaxCoalescedDistanceBytes] =
             std::to_string(FLAGS_max_coalesced_distance_bytes);
+    configurationValues[connector::hive::HiveConfig::kPrefetchRowGroups] =
+        std::to_string(FLAGS_parquet_prefetch_rowgroups);
     auto properties =
         std::make_shared<const core::MemConfig>(configurationValues);