Skip to content

Commit f6d42ed

Browse files
committed
function engine
1 parent ff67475 commit f6d42ed

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

cpp/velox/compute/VeloxBackend.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
#include "jni/JniFileSystem.h"
4040
#include "memory/GlutenBufferedInputBuilder.h"
4141
#include "operators/functions/SparkExprToSubfieldFilterParser.h"
42+
#include "operators/plannodes/RowVectorStream.h"
4243
#include "shuffle/ArrowShuffleDictionaryWriter.h"
4344
#include "udf/UdfLoader.h"
4445
#include "utils/Exception.h"
@@ -47,7 +48,6 @@
4748
#include "velox/connectors/hive/BufferedInputBuilder.h"
4849
#include "velox/connectors/hive/HiveConnector.h"
4950
#include "velox/connectors/hive/HiveDataSource.h"
50-
#include "operators/plannodes/RowVectorStream.h"
5151
#include "velox/connectors/hive/storage_adapters/abfs/RegisterAbfsFileSystem.h" // @manual
5252
#include "velox/connectors/hive/storage_adapters/gcs/RegisterGcsFileSystem.h" // @manual
5353
#include "velox/connectors/hive/storage_adapters/hdfs/HdfsFileSystem.h"
@@ -177,7 +177,8 @@ void VeloxBackend::init(
177177
{velox::cudf_velox::CudfConfig::kCudfMemoryResource,
178178
backendConf_->get(kCudfMemoryResource, kCudfMemoryResourceDefault)},
179179
{velox::cudf_velox::CudfConfig::kCudfMemoryPercent,
180-
backendConf_->get(kCudfMemoryPercent, kCudfMemoryPercentDefault)}};
180+
backendConf_->get(kCudfMemoryPercent, kCudfMemoryPercentDefault)},
181+
{velox::cudf_velox::CudfConfig::kCudfFunctionEngine, "spark"}};
181182
auto& cudfConfig = velox::cudf_velox::CudfConfig::getInstance();
182183
cudfConfig.initialize(std::move(options));
183184
velox::cudf_velox::registerCudf();
@@ -317,13 +318,13 @@ void VeloxBackend::initConnector(const std::shared_ptr<velox::config::ConfigBase
317318
}
318319
velox::connector::registerConnector(
319320
std::make_shared<velox::connector::hive::HiveConnector>(kHiveConnectorId, hiveConf, ioExecutor_.get()));
320-
321+
321322
// Register value-stream connector for runtime iterator-based inputs
322323
auto valueStreamDynamicFilterEnabled =
323324
backendConf_->get<bool>(kValueStreamDynamicFilterEnabled, kValueStreamDynamicFilterEnabledDefault);
324325
velox::connector::registerConnector(
325326
std::make_shared<ValueStreamConnector>(kIteratorConnectorId, hiveConf, valueStreamDynamicFilterEnabled));
326-
327+
327328
#ifdef GLUTEN_ENABLE_GPU
328329
if (backendConf_->get<bool>(kCudfEnableTableScan, kCudfEnableTableScanDefault) &&
329330
backendConf_->get<bool>(kCudfEnabled, kCudfEnabledDefault)) {

0 commit comments

Comments
 (0)