Package org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu

Class GpuDiscoverer

java.lang.Object

org.apache.hadoop.conf.Configured

org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu.GpuDiscoverer

All Implemented Interfaces:

org.apache.hadoop.conf.Configurable

@Private
@Unstable
public class GpuDiscoverer
extends org.apache.hadoop.conf.Configured

Field Summary

Fields

Modifier and Type	Field	Description
static org.slf4j.Logger	LOG

Constructor Summary

Constructors

Constructor	Description
GpuDiscoverer()

Method Summary

Modifier and Type	Method	Description
GpuDeviceInformation	getGpuDeviceInformation()	Get GPU device information from system.
java.util.List<GpuDevice>	getGpusUsableByYarn()	Get list of GPU devices usable by YARN.
void	initialize(org.apache.hadoop.conf.Configuration config, NvidiaBinaryHelper nvidiaHelper)

Methods inherited from class org.apache.hadoop.conf.Configured

getConf, setConf

Methods inherited from class java.lang.Object

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

Field Detail

LOG

public static final org.slf4j.Logger LOG

Constructor Detail

GpuDiscoverer

public GpuDiscoverer()

Method Detail

getGpuDeviceInformation

public GpuDeviceInformation getGpuDeviceInformation()
                                             throws org.apache.hadoop.yarn.exceptions.YarnException

Get GPU device information from system. This need to be called after initialize. Please note that this only works on *NIX platform, so external caller need to make sure this.

Returns:

GpuDeviceInformation

Throws:

org.apache.hadoop.yarn.exceptions.YarnException - when any error happens

getGpusUsableByYarn

public java.util.List<GpuDevice> getGpusUsableByYarn()
                                              throws org.apache.hadoop.yarn.exceptions.YarnException

Get list of GPU devices usable by YARN.

Returns:

List of GPU devices

Throws:

org.apache.hadoop.yarn.exceptions.YarnException

initialize

public void initialize(org.apache.hadoop.conf.Configuration config,
                       NvidiaBinaryHelper nvidiaHelper)
                throws org.apache.hadoop.yarn.exceptions.YarnException

Throws:

org.apache.hadoop.yarn.exceptions.YarnException