ml.dmlc.xgboost4j.java

Class XGBoostSparkJNI

java.lang.Object

ml.dmlc.xgboost4j.java.XGBoostSparkJNI

public class XGBoostSparkJNI
extends Object

JNI functions for XGBoost4J-Spark

Constructor Summary

Constructors

Constructor and Description
XGBoostSparkJNI()

Method Summary

Modifier and Type	Method and Description
static int	allocateGpuDevice()
static long	buildUnsafeRows(long[] nativeColumnPtrs) Build an array of fixed-length Spark UnsafeRow using the GPU.
static int	getGpuDevice()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

XGBoostSparkJNI

public XGBoostSparkJNI()

Method Detail

buildUnsafeRows

public static long buildUnsafeRows(long[] nativeColumnPtrs)

Build an array of fixed-length Spark UnsafeRow using the GPU.

Parameters:

nativeColumnPtrs - native address of cudf column pointers

Returns:

native address of the UnsafeRow array NOTE: It is the responsibility of the caller to free the native memory returned by this function (e.g.: using Platform.freeMemory).

getGpuDevice

public static int getGpuDevice()

allocateGpuDevice

public static int allocateGpuDevice()