| 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 |
AssignedGpuDevice
In 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–2024 Apache Software Foundation. All rights reserved.