org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.resources.fpga

Class FpgaResourceAllocator

java.lang.Object

org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.resources.fpga.FpgaResourceAllocator

public class FpgaResourceAllocator
extends Object

This FPGA resource allocator tends to be used by different FPGA vendor's plugin A "type" parameter is taken into consideration when allocation

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FpgaResourceAllocator.FpgaAllocation

Constructor Summary

Constructors

Constructor and Description
FpgaResourceAllocator(Context ctx)

Method Summary

Modifier and Type	Method and Description
void	addFpgaDevices(String type, List<FpgaDevice> list)
FpgaResourceAllocator.FpgaAllocation	assignFpga(String type, long count, Container container, String ipidHash) Assign FpgaResourceAllocator.FpgaAllocation with preferred IPID, if no, with random FPGAs
void	cleanupAssignFpgas(String requestor)
void	recoverAssignedFpgas(org.apache.hadoop.yarn.api.records.ContainerId containerId)
void	updateFpga(String requestor, FpgaDevice device, String newIPID, String newHash)

Methods inherited from class java.lang.Object

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

Constructor Detail

FpgaResourceAllocator

public FpgaResourceAllocator(Context ctx)

Method Detail

addFpgaDevices

public void addFpgaDevices(String type,
                           List<FpgaDevice> list)

updateFpga

public void updateFpga(String requestor,
                       FpgaDevice device,
                       String newIPID,
                       String newHash)

assignFpga

public FpgaResourceAllocator.FpgaAllocation assignFpga(String type,
                                                       long count,
                                                       Container container,
                                                       String ipidHash)
                                                throws ResourceHandlerException

Assign FpgaResourceAllocator.FpgaAllocation with preferred IPID, if no, with random FPGAs

Parameters:

type - vendor plugin supported FPGA device type

count - requested FPGA slot count

container - container id

ipidHash - hash of the localized aocx file

Returns:

Instance consists two List of allowed and denied FpgaDevice

Throws:

ResourceHandlerException - When failed to allocate or write state store

recoverAssignedFpgas

public void recoverAssignedFpgas(org.apache.hadoop.yarn.api.records.ContainerId containerId)
                          throws ResourceHandlerException

Throws:

ResourceHandlerException

cleanupAssignFpgas

public void cleanupAssignFpgas(String requestor)