Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[jvm-packages] fix spark-rapids compatibility issue (#8240)

Browse Source 
* [jvm-packages] fix spark-rapids compatibility issue

spark-rapids (from 22.10) has shimmed GpuColumnVector, which means
we can't call it directly. So this PR call the UnshimmedGpuColumnVector
This commit is contained in:
Bobby Wang
2022-09-22 23:31:29 +08:00
committed by GitHub
parent ab342af242
commit 8d247f0d64
4 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/util/Utils.scala
View File

@@ -19,7 +19,7 @@ package ml.dmlc.xgboost4j.scala.spark.util
import org.json4s.{DefaultFormats, FullTypeHints, JField, JValue, NoTypeHints, TypeHints}
// based on org.apache.spark.util copy /paste
private[spark] object Utils {
object Utils {
def getSparkClassLoader: ClassLoader = getClass.getClassLoader