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