@InterfaceAudience.Private @InterfaceStability.Unstable public class GpuDiscoverer extends Object
Modifier and Type | Field and Description |
---|---|
protected static String |
DEFAULT_BINARY_NAME |
static org.slf4j.Logger |
LOG |
Constructor and Description |
---|
GpuDiscoverer() |
Modifier and Type | Method and Description |
---|---|
protected Map<String,String> |
getEnvironmentToRunCommand() |
GpuDeviceInformation |
getGpuDeviceInformation()
Get GPU device information from system.
|
List<GpuDevice> |
getGpusUsableByYarn()
Get list of GPU devices usable by YARN.
|
static GpuDiscoverer |
getInstance() |
protected String |
getPathOfGpuBinary() |
void |
initialize(org.apache.hadoop.conf.Configuration conf) |
public static final org.slf4j.Logger LOG
protected static final String DEFAULT_BINARY_NAME
public GpuDeviceInformation getGpuDeviceInformation() throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnException
- when any error happenspublic List<GpuDevice> getGpusUsableByYarn() throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnException
- when any issue happenspublic void initialize(org.apache.hadoop.conf.Configuration conf) throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnException
protected String getPathOfGpuBinary()
public static GpuDiscoverer getInstance()
Copyright © 2022 Apache Software Foundation. All rights reserved.