|
12 | 12 | * limitations under the License. |
13 | 13 | */ |
14 | 14 | #include "presto_cpp/main/connectors/Registration.h" |
15 | | -#include "presto_cpp/main/connectors/IcebergPrestoToVeloxConnector.h" |
16 | 15 | #include "presto_cpp/main/connectors/HivePrestoToVeloxConnector.h" |
| 16 | +#include "presto_cpp/main/connectors/IcebergPrestoToVeloxConnector.h" |
17 | 17 | #include "presto_cpp/main/connectors/SystemConnector.h" |
18 | 18 |
|
19 | 19 | #ifdef PRESTO_ENABLE_ARROW_FLIGHT_CONNECTOR |
@@ -91,6 +91,21 @@ void registerConnectorFactories() { |
91 | 91 | facebook::presto::registerConnectorFactory( |
92 | 92 | std::make_shared<facebook::velox::connector::hive::HiveConnectorFactory>( |
93 | 93 | kHiveHadoop2ConnectorName)); |
| 94 | +#ifdef PRESTO_ENABLE_CUDF |
| 95 | + facebook::presto::unregisterConnectorFactory( |
| 96 | + facebook::velox::connector::hive::HiveConnectorFactory::kHiveConnectorName); |
| 97 | + facebook::presto::unregisterConnectorFactory(kHiveHadoop2ConnectorName); |
| 98 | + |
| 99 | + // Register cuDF Hive connector factory |
| 100 | + facebook::presto::registerConnectorFactory( |
| 101 | + std::make_shared<facebook::velox::cudf_velox::connector::hive:: |
| 102 | + CudfHiveConnectorFactory>()); |
| 103 | + |
| 104 | + // Register cudf Hive connector factory |
| 105 | + facebook::presto::registerConnectorFactory( |
| 106 | + std::make_shared<facebook::velox::cudf_velox::connector::hive:: |
| 107 | + CudfHiveConnectorFactory>(kHiveHadoop2ConnectorName)); |
| 108 | +#endif |
94 | 109 |
|
95 | 110 | // Register TPC-DS connector factory |
96 | 111 | facebook::presto::registerConnectorFactory( |
|
0 commit comments