Package org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.deviceframework

Class DeviceMappingManager

java.lang.Object

org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.deviceframework.DeviceMappingManager

public class DeviceMappingManager
extends java.lang.Object

Schedule device resource based on requirements and do book keeping It holds all device type resource and can do scheduling as a default scheduler.

Constructor Summary

Constructors

Constructor	Description
DeviceMappingManager(Context context)

Method Summary

Modifier and Type	Method	Description
void	addDevicePluginScheduler(java.lang.String resourceName, DevicePluginScheduler s)
void	addDeviceSet(java.lang.String resourceName, java.util.Set<Device> deviceSet)
org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.deviceframework.DeviceMappingManager.DeviceAllocation	assignDevices(java.lang.String resourceName, Container container)
void	cleanupAssignedDevices(java.lang.String resourceName, org.apache.hadoop.yarn.api.records.ContainerId containerId)
java.util.Map<java.lang.String,java.util.Set<Device>>	getAllAllowedDevices()
java.util.Set<Device>	getAllocatedDevices(java.lang.String resourceName, org.apache.hadoop.yarn.api.records.ContainerId cId)
java.util.Map<java.lang.String,java.util.Map<Device,org.apache.hadoop.yarn.api.records.ContainerId>>	getAllUsedDevices()
int	getAvailableDevices(java.lang.String resourceName)
java.util.Map<java.lang.String,DevicePluginScheduler>	getDevicePluginSchedulers()
static int	getRequestedDeviceCount(java.lang.String resName, org.apache.hadoop.yarn.api.records.Resource requestedResource)
void	recoverAssignedDevices(java.lang.String resourceName, org.apache.hadoop.yarn.api.records.ContainerId containerId)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DeviceMappingManager

public DeviceMappingManager(Context context)

Method Detail

getAllAllowedDevices

@VisibleForTesting
public java.util.Map<java.lang.String,java.util.Set<Device>> getAllAllowedDevices()

getAllUsedDevices

@VisibleForTesting
public java.util.Map<java.lang.String,java.util.Map<Device,org.apache.hadoop.yarn.api.records.ContainerId>> getAllUsedDevices()

getDevicePluginSchedulers

@VisibleForTesting
public java.util.Map<java.lang.String,DevicePluginScheduler> getDevicePluginSchedulers()

getAllocatedDevices

@VisibleForTesting
public java.util.Set<Device> getAllocatedDevices(java.lang.String resourceName,
                                                 org.apache.hadoop.yarn.api.records.ContainerId cId)

addDeviceSet

public void addDeviceSet(java.lang.String resourceName,
                         java.util.Set<Device> deviceSet)

assignDevices

public org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.deviceframework.DeviceMappingManager.DeviceAllocation assignDevices(java.lang.String resourceName,
                                                                                                                                                     Container container)
                                                                                                                                              throws ResourceHandlerException

Throws:

ResourceHandlerException

recoverAssignedDevices

public void recoverAssignedDevices(java.lang.String resourceName,
                                   org.apache.hadoop.yarn.api.records.ContainerId containerId)
                            throws ResourceHandlerException

Throws:

ResourceHandlerException

cleanupAssignedDevices

public void cleanupAssignedDevices(java.lang.String resourceName,
                                   org.apache.hadoop.yarn.api.records.ContainerId containerId)

getRequestedDeviceCount

public static int getRequestedDeviceCount(java.lang.String resName,
                                          org.apache.hadoop.yarn.api.records.Resource requestedResource)

getAvailableDevices

public int getAvailableDevices(java.lang.String resourceName)

addDevicePluginScheduler

@VisibleForTesting
public void addDevicePluginScheduler(java.lang.String resourceName,
                                     DevicePluginScheduler s)