@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.YarnExceptionprotected String getPathOfGpuBinary()
public static GpuDiscoverer getInstance()
Copyright © 2022 Apache Software Foundation. All rights reserved.