Update ColumnarToRowBenchmark.cc

apache · Aug 1, 2024 · c4f22f6 · c4f22f6
1 parent 6802015
commit c4f22f6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cpp/velox/benchmarks/ColumnarToRowBenchmark.cc b/cpp/velox/benchmarks/ColumnarToRowBenchmark.cc
@@ -110,7 +110,7 @@ class GoogleBenchmarkColumnarToRow {
 };
 class GoogleBenchmarkColumnarToRowCacheScanBenchmark : public GoogleBenchmarkColumnarToRow {
  public:
-  GoogleBenchmarkColumnarToRowCacheScanBenchmark(std::string filename) : GoogleBenchmarkColumnarToRow(filename, 64<<20) {}
+  GoogleBenchmarkColumnarToRowCacheScanBenchmark(std::string filename) : GoogleBenchmarkColumnarToRow(filename) {}
   void operator()(benchmark::State& state) {
     if (state.range(0) == 0xffffffff) {
       setCpu(state.thread_index());
@@ -157,7 +157,7 @@ class GoogleBenchmarkColumnarToRowCacheScanBenchmark : public GoogleBenchmarkCol
     for (auto _ : state) {
       for (const auto& vector : vectors) {
         auto row = std::dynamic_pointer_cast<velox::RowVector>(vector);
-        auto columnarToRowConverter = std::make_shared<gluten::VeloxColumnarToRowConverter>(ctxPool);
+        auto columnarToRowConverter = std::make_shared<gluten::VeloxColumnarToRowConverter>(ctxPool, 64 << 20);
         auto cb = std::make_shared<VeloxColumnarBatch>(row);
         TIME_NANO_START(writeTime);
         columnarToRowConverter->convert(cb);
@@ -212,7 +212,7 @@ class GoogleBenchmarkColumnarToRowIterateScanBenchmark : public GoogleBenchmarkC
         numBatches += 1;
         numRows += recordBatch->num_rows();
         auto vector = recordBatch2RowVector(*recordBatch);
-        auto columnarToRowConverter = std::make_shared<gluten::VeloxColumnarToRowConverter>(ctxPool, 64<<20);
+        auto columnarToRowConverter = std::make_shared<gluten::VeloxColumnarToRowConverter>(ctxPool, 64 << 20);
         auto row = std::dynamic_pointer_cast<velox::RowVector>(vector);
         auto cb = std::make_shared<VeloxColumnarBatch>(row);
         TIME_NANO_START(writeTime);