Skip to content

Commit e794084

Browse files
[Minor] register the Spark functions in GPU
1 parent 75b4323 commit e794084

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

cpp/velox/compute/VeloxBackend.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,9 @@
3232
#include "operators/plannodes/CudfVectorStream.h"
3333
#include "velox/experimental/cudf/CudfConfig.h"
3434
#include "velox/experimental/cudf/connectors/hive/CudfHiveConnector.h"
35+
#include "velox/experimental/cudf/exec/SparkAggregateFunctions.h"
3536
#include "velox/experimental/cudf/exec/ToCudf.h"
37+
#include "velox/experimental/cudf/expression/SparkFunctions.h"
3638
#endif
3739

3840
#include "compute/VeloxRuntime.h"
@@ -184,6 +186,8 @@ void VeloxBackend::init(
184186
cudfConfig.initialize(std::move(options));
185187
velox::cudf_velox::registerCudf();
186188
velox::exec::Operator::registerOperator(std::make_unique<CudfVectorStreamOperatorTranslator>());
189+
velox::cudf_velox::registerSparkFunctions("");
190+
velox::cudf_velox::registerSparkAggregateFunctions("");
187191
}
188192
#endif
189193

0 commit comments

Comments
 (0)