hex.tree.xgboost.util

Class GpuUtils

java.lang.Object

hex.tree.xgboost.util.GpuUtils

public class GpuUtils
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
static int[]	DEFAULT_GPU_ID

Constructor Summary

Constructors

Constructor and Description
GpuUtils()

Method Summary

Modifier and Type	Method and Description
static java.util.Set<java.lang.Integer>	allGPUs()
static java.util.Set<java.lang.Integer>	allGPUs(water.H2ONode node)
static boolean	hasGPU()
static boolean	hasGPU(water.H2ONode node, int[] gpu_id)
static boolean	hasGPU(int[] gpu_id)
static int	numGPUs(water.H2ONode node)

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_GPU_ID

public static final int[] DEFAULT_GPU_ID

Constructor Detail

GpuUtils

public GpuUtils()

Method Detail

numGPUs

public static int numGPUs(water.H2ONode node)

allGPUs

public static java.util.Set<java.lang.Integer> allGPUs(water.H2ONode node)

allGPUs

public static java.util.Set<java.lang.Integer> allGPUs()

hasGPU

public static boolean hasGPU(water.H2ONode node,
                             int[] gpu_id)

hasGPU

public static boolean hasGPU(int[] gpu_id)

hasGPU

public static boolean hasGPU()