| Package | Description | 
|---|---|
| org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.resources.gpu | |
| org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu | Package for GPU support classes. | 
| org.apache.hadoop.yarn.server.nodemanager.webapp.dao.gpu | 
| Modifier and Type | Method and Description | 
|---|---|
| List<GpuDevice> | GpuResourceAllocator. getAllowedGpus() | 
| Map<GpuDevice,org.apache.hadoop.yarn.api.records.ContainerId> | GpuResourceAllocator. getDeviceAllocationMapping() | 
| Modifier and Type | Method and Description | 
|---|---|
| void | GpuResourceAllocator. addGpu(GpuDevice gpuDevice)Add GPU to the allowed list of GPUs. | 
| Modifier and Type | Class and Description | 
|---|---|
| class  | AssignedGpuDeviceIn addition to  GpuDevice, this include container id and more runtime
 information related to who is using the GPU device if possible | 
| Modifier and Type | Method and Description | 
|---|---|
| List<GpuDevice> | GpuDiscoverer. getGpusUsableByYarn()Get list of GPU devices usable by YARN. | 
| Modifier and Type | Method and Description | 
|---|---|
| List<GpuDevice> | NMGpuResourceInfo. getTotalGpuDevices() | 
| Modifier and Type | Method and Description | 
|---|---|
| void | NMGpuResourceInfo. setTotalGpuDevices(List<GpuDevice> totalGpuDevices) | 
| Constructor and Description | 
|---|
| NMGpuResourceInfo(GpuDeviceInformation gpuDeviceInformation,
                 List<GpuDevice> totalGpuDevices,
                 List<AssignedGpuDevice> assignedGpuDevices) | 
Copyright © 2008–2023 Apache Software Foundation. All rights reserved.