public class GpuResourceAllocator extends Object
Constructor and Description |
---|
GpuResourceAllocator(Context ctx) |
Modifier and Type | Method and Description |
---|---|
void |
addGpu(GpuDevice gpuDevice)
Add GPU to the allowed list of GPUs.
|
org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.resources.gpu.GpuResourceAllocator.GpuAllocation |
assignGpus(Container container)
Assign GPU to the specified container.
|
List<GpuDevice> |
getAllowedGpus() |
List<AssignedGpuDevice> |
getAssignedGpus() |
int |
getAvailableGpus() |
Map<GpuDevice,org.apache.hadoop.yarn.api.records.ContainerId> |
getDeviceAllocationMapping() |
static int |
getRequestedGpus(org.apache.hadoop.yarn.api.records.Resource requestedResource)
Get number of requested GPUs from resource.
|
void |
recoverAssignedGpus(org.apache.hadoop.yarn.api.records.ContainerId containerId) |
String |
toString() |
void |
unassignGpus(org.apache.hadoop.yarn.api.records.ContainerId containerId)
Clean up all GPUs assigned to containerId.
|
public GpuResourceAllocator(Context ctx)
public void addGpu(GpuDevice gpuDevice)
gpuDevice
- gpu devicepublic int getAvailableGpus()
public void recoverAssignedGpus(org.apache.hadoop.yarn.api.records.ContainerId containerId) throws ResourceHandlerException
ResourceHandlerException
public static int getRequestedGpus(org.apache.hadoop.yarn.api.records.Resource requestedResource)
requestedResource
- requested resourcepublic org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.resources.gpu.GpuResourceAllocator.GpuAllocation assignGpus(Container container) throws ResourceHandlerException
container
- container to allocateResourceHandlerException
- When failed to assign GPUs.public void unassignGpus(org.apache.hadoop.yarn.api.records.ContainerId containerId)
containerId
- containerIdpublic Map<GpuDevice,org.apache.hadoop.yarn.api.records.ContainerId> getDeviceAllocationMapping()
public List<AssignedGpuDevice> getAssignedGpus()
Copyright © 2008–2023 Apache Software Foundation. All rights reserved.