Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer.MemoryPlatform

class Deeploy.MemoryLevelExtension.NetworkDeployers.MemoryLevelDeployer.MemoryPlatform(memoryHierarchy: MemoryHierarchy, defaultTargetMemoryLevel: MemoryLevel, engines: List[DeploymentEngine], variableBuffer: Type[VariableBuffer], constantBuffer: Type[ConstantBuffer], structBuffer: Type[StructBuffer], transientBuffer: Type[TransientBuffer])

Bases: DeploymentPlatform

Methods

__init__(memoryHierarchy: MemoryHierarchy, defaultTargetMemoryLevel: MemoryLevel, engines: List[DeploymentEngine], variableBuffer: Type[VariableBuffer], constantBuffer: Type[ConstantBuffer], structBuffer: Type[StructBuffer], transientBuffer: Type[TransientBuffer]) → None

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__(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]