From cb71a1c78aafed62ee1744d96ecd6bdfa3a2ff65 Mon Sep 17 00:00:00 2001
From: PHILO-HE <feilong.he@intel.com>
Date: Mon, 16 Dec 2024 18:18:19 +0800
Subject: [PATCH] Initial

---
 velox/exec/benchmarks/WindowPrefixSortBenchmark.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/velox/exec/benchmarks/WindowPrefixSortBenchmark.cpp b/velox/exec/benchmarks/WindowPrefixSortBenchmark.cpp
index c91f40a490f66..43b8d5ee10129 100644
--- a/velox/exec/benchmarks/WindowPrefixSortBenchmark.cpp
+++ b/velox/exec/benchmarks/WindowPrefixSortBenchmark.cpp
@@ -19,6 +19,7 @@
 #include <folly/init/Init.h>
 #include <string>
 
+#include "velox/common/memory/SharedArbitrator.h"
 #include "velox/exec/Cursor.h"
 #include "velox/exec/tests/utils/HiveConnectorTestBase.h"
 #include "velox/exec/tests/utils/PlanBuilder.h"
@@ -41,6 +42,7 @@ namespace {
 class WindowPrefixSortBenchmark : public HiveConnectorTestBase {
  public:
   explicit WindowPrefixSortBenchmark() {
+    memory::SharedArbitrator::registerFactory();
     HiveConnectorTestBase::SetUp();
     aggregate::prestosql::registerAllAggregateFunctions();
     window::prestosql::registerAllWindowFunctions();