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

Class GpuDiscoverer

java.lang.Object

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

@InterfaceAudience.Private
 @InterfaceStability.Unstable
public class GpuDiscoverer
extends Object

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor and Description
GpuDiscoverer()

Method Summary

Modifier and Type	Method and Description
List<GpuDevice>	getGpusUsableByYarn() Get list of GPU devices usable by YARN.
void	initialize(org.apache.hadoop.conf.Configuration config)

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

getGpusUsableByYarn

public 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 - when any issue happens

initialize

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

Throws:

org.apache.hadoop.yarn.exceptions.YarnException