Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer.MemoryPlatformWrapper

class Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer.MemoryPlatformWrapper(platform: DeploymentPlatform, memoryHierarchy: MemoryHierarchy, defaultTargetMemoryLevel: MemoryLevel)

Bases: DeploymentPlatformWrapper

Methods

__init__(platform: DeploymentPlatform, memoryHierarchy: MemoryHierarchy, defaultTargetMemoryLevel: MemoryLevel)

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__(platform, memoryHierarchy, ...)

Initializes a new deployment platform

getTargetMemoryLevel(node, tensorName, ctxt)

Attributes

engines

A list of all available non-host engines
engines

A list of all available non-host engines

Type:

List[DeploymentEngine]