File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 1919
2020#include " arrow/array/array_base.h"
2121#include " arrow/compute/api.h"
22+ #include " arrow/compute/exec_internal.h"
2223#include " arrow/memory_pool.h"
2324#include " arrow/scalar.h"
2425#include " arrow/testing/gtest_util.h"
@@ -179,16 +180,17 @@ void BM_ExecBatchIterator(benchmark::State& state) {
179180 random::RandomArrayGenerator rag (kSeed );
180181
181182 const int64_t length = 1 << 20 ;
182- const int num_fields = 10 ;
183+ const int num_fields = 32 ;
183184
184185 std::vector<Datum> args (num_fields);
185186 for (int i = 0 ; i < num_fields; ++i) {
186187 args[i] = rag.Int64 (length, 0 , 100 )->data ();
187188 }
188189
190+ const int64_t blocksize = state.range (0 );
189191 for (auto _ : state) {
190- std::unique_ptr<ExecBatchIterator> it =
191- * ExecBatchIterator::Make (args, state. range ( 0 ) );
192+ std::unique_ptr<detail:: ExecBatchIterator> it =
193+ * detail:: ExecBatchIterator::Make (args, blocksize );
192194 ExecBatch batch;
193195 while (it->Next (&batch)) {
194196 for (int i = 0 ; i < num_fields; ++i) {
You can’t perform that action at this time.
0 commit comments