[jvm-packages] Add Rapids plugin support (#7491)

* Add GPU pre-processing pipeline.
2021-12-17 13:11:12 +08:00
parent 5b1161bb64
commit 24e25802a7
24 changed files with 2035 additions and 37 deletions
--- a/jvm-packages/xgboost4j-spark-gpu/pom.xml
+++ b/jvm-packages/xgboost4j-spark-gpu/pom.xml
@@ -44,5 +44,18 @@
            <version>${spark.version}</version>
            <scope>provided</scope>
        </dependency>
+        <dependency>
+          <groupId>ai.rapids</groupId>
+          <artifactId>cudf</artifactId>
+          <version>${cudf.version}</version>
+          <classifier>${cudf.classifier}</classifier>
+          <scope>provided</scope>
+        </dependency>
+        <dependency>
+          <groupId>com.nvidia</groupId>
+          <artifactId>rapids-4-spark_${scala.binary.version}</artifactId>
+          <version>${spark.rapids.version}</version>
+          <scope>provided</scope>
+        </dependency>
    </dependencies>
 </project>