Deeploy.Targets.MemPool.Platform.MemPoolPlatform

class Deeploy.Targets.MemPool.Platform.MemPoolPlatform(engines=[MemPoolEngine(name='MemPool', mappings=['Add', 'Conv', 'DebugPrint', 'Div', 'Flatten', 'Gather', 'Gemm', 'iGELU', 'iLayerNorm', 'IntegerDiv', 'IntegerMean', 'iSoftmax', 'ITAMax', 'MatMul', 'MatMulInteger', 'MaxPool', 'MHSA', 'Mul', 'Pad', 'ReduceMean', 'ReduceSum', 'RequantizediGELU', 'RequantShift', 'Reshape', 'RQIntegerDiv', 'RQGemm', 'RQMatMul', 'Transpose', 'Unsqueeze', 'Slice'])], variableBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolVariableBuffer'>, constantBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolConstantBuffer'>, structBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolStructBuffer'>, transientBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolTransientBuffer'>)

Bases: DeploymentPlatform

Methods

__init__(engines=[MemPoolEngine(name='MemPool', mappings=['Add', 'Conv', 'DebugPrint', 'Div', 'Flatten', 'Gather', 'Gemm', 'iGELU', 'iLayerNorm', 'IntegerDiv', 'IntegerMean', 'iSoftmax', 'ITAMax', 'MatMul', 'MatMulInteger', 'MaxPool', 'MHSA', 'Mul', 'Pad', 'ReduceMean', 'ReduceSum', 'RequantizediGELU', 'RequantShift', 'Reshape', 'RQIntegerDiv', 'RQGemm', 'RQMatMul', 'Transpose', 'Unsqueeze', 'Slice'])], variableBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolVariableBuffer'>, constantBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolConstantBuffer'>, structBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolStructBuffer'>, transientBuffer=<class 'Deeploy.Targets.MemPool.Platform.MemPoolTransientBuffer'>)

Initializes a new deployment platform

Parameters:

  • engines (List[DeploymentEngine]) – List of all available non-host engines

  • variableBuffer (Type[VariableBuffer]) – VariableBuffer subclass with correctly set allocation and deallocation templates

  • constantBuffer (Type[ConstantBuffer]) – ConstantBuffer subclass with correctly set allocation and deallocation templates

  • structBuffer (Type[StructBuffer]) – StructBuffer subclass with correctly set allocation and deallocation templates

  • transientBuffer (Type[TransientBuffer]) – TransientBuffer subclass with correctly set allocation and deallocation templates

__init__([engines, variableBuffer, ...])

Initializes a new deployment platform

Attributes

engines

A list of all available non-host engines
engines

A list of all available non-host engines

Type:

List[DeploymentEngine]