[jvm-packages] rework xgboost4j-spark and xgboost4j-spark-gpu

jvm-packages/xgboost4j-spark-gpu/src/main/java/ml/dmlc/xgboost4j/java/CudfColumn.java

@@ -55,9 +55,9 @@ public static CudfColumn from(ColumnVector cv) {
     DType dType = cv.getType();
     String typeStr = "";
     if (dType == DType.FLOAT32 || dType == DType.FLOAT64 ||
-          dType == DType.TIMESTAMP_DAYS || dType == DType.TIMESTAMP_MICROSECONDS ||
-          dType == DType.TIMESTAMP_MILLISECONDS || dType == DType.TIMESTAMP_NANOSECONDS ||
-          dType == DType.TIMESTAMP_SECONDS) {
+        dType == DType.TIMESTAMP_DAYS || dType == DType.TIMESTAMP_MICROSECONDS ||
+        dType == DType.TIMESTAMP_MILLISECONDS || dType == DType.TIMESTAMP_NANOSECONDS ||
+        dType == DType.TIMESTAMP_SECONDS) {
       typeStr = "<f" + dType.getSizeInBytes();
     } else if (dType == DType.BOOL8 || dType == DType.INT8 || dType == DType.INT16 ||
         dType == DType.INT32 || dType == DType.INT64) {

jvm-packages/xgboost4j-spark-gpu/src/main/resources/META-INF/services/ml.dmlc.xgboost4j.scala.spark.XGBoostPlugin

@@ -0,0 +1 @@
+ml.dmlc.xgboost4j.scala.spark.GpuXGBoostPlugin

jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/QuantileDMatrix.scala

@@ -0,0 +1,112 @@
+/*
+ Copyright (c) 2021-2024 by Contributors
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ */
+
+package ml.dmlc.xgboost4j.scala
+
+import _root_.scala.collection.JavaConverters._
+
+import ml.dmlc.xgboost4j.java.{Column, ColumnBatch, QuantileDMatrix => JQuantileDMatrix, XGBoostError}
+
+class QuantileDMatrix private[scala](
+    private[scala] override val jDMatrix: JQuantileDMatrix) extends DMatrix(jDMatrix) {
+
+  /**
+   * Create QuantileDMatrix from iterator based on the array interface
+   *
+   * @param iter    the XGBoost ColumnBatch batch to provide the corresponding array interface
+   * @param missing the missing value
+   * @param maxBin  the max bin
+   * @param nthread the parallelism
+   * @throws XGBoostError
+   */
+  def this(iter: Iterator[ColumnBatch], missing: Float, maxBin: Int, nthread: Int) {
+    this(new JQuantileDMatrix(iter.asJava, missing, maxBin, nthread))
+  }
+
+  /**
+   * set label of dmatrix
+   *
+   * @param labels labels
+   */
+  @throws(classOf[XGBoostError])
+  override def setLabel(labels: Array[Float]): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setLabel.")
+
+  /**
+   * set weight of each instance
+   *
+   * @param weights weights
+   */
+  @throws(classOf[XGBoostError])
+  override def setWeight(weights: Array[Float]): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setWeight.")
+
+  /**
+   * if specified, xgboost will start from this init margin
+   * can be used to specify initial prediction to boost from
+   *
+   * @param baseMargin base margin
+   */
+  @throws(classOf[XGBoostError])
+  override def setBaseMargin(baseMargin: Array[Float]): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setBaseMargin.")
+
+  /**
+   * if specified, xgboost will start from this init margin
+   * can be used to specify initial prediction to boost from
+   *
+   * @param baseMargin base margin
+   */
+  @throws(classOf[XGBoostError])
+  override def setBaseMargin(baseMargin: Array[Array[Float]]): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setBaseMargin.")
+
+  /**
+   * Set group sizes of DMatrix (used for ranking)
+   *
+   * @param group group size as array
+   */
+  @throws(classOf[XGBoostError])
+  override def setGroup(group: Array[Int]): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setGroup.")
+
+  /**
+   * Set label of DMatrix from array interface
+   */
+  @throws(classOf[XGBoostError])
+  override def setLabel(column: Column): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setLabel.")
+
+  /**
+   * set weight of dmatrix from column array interface
+   */
+  @throws(classOf[XGBoostError])
+  override def setWeight(column: Column): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setWeight.")
+
+  /**
+   * set base margin of dmatrix from column array interface
+   */
+  @throws(classOf[XGBoostError])
+  override def setBaseMargin(column: Column): Unit =
+    throw new XGBoostError("QuantileDMatrix does not support setBaseMargin.")
+
+  @throws(classOf[XGBoostError])
+  override def setQueryId(column: Column): Unit = {
+    throw new XGBoostError("QuantileDMatrix does not support setQueryId.")
+  }
+
+}