Deeploy.Targets.Snitch.Platform.SnitchClusterEngine

class Deeploy.Targets.Snitch.Platform.SnitchClusterEngine(name: str, Mapping={'Add': <Deeploy.Targets.Generic.Layers.AddLayer object>, 'Gather': <Deeploy.Targets.Generic.Layers.GatherLayer object>, 'Gemm': <Deeploy.Targets.Generic.Layers.GEMMLayer object>, 'MatMul': <Deeploy.Targets.Generic.Layers.MatMulLayer object>, 'Pad': <Deeploy.Targets.Generic.Layers.PadLayer object>, 'RQGemm': <Deeploy.Targets.Generic.Layers.RQGEMMLayer object>, 'RQIntegerDiv': <Deeploy.Targets.Generic.Layers.RQIntegerDivLayer object>, 'RequantizedAdd': <Deeploy.Targets.Generic.Layers.AddLayer object>, 'Softmax': <Deeploy.Targets.Generic.Layers.SoftmaxLayer object>, 'Unsqueeze': <Deeploy.Targets.Generic.Layers.ReshapeLayer object>, 'iLayerNorm': <Deeploy.Targets.Generic.Layers.LayerNormLayer object>, 'iNoNorm': <Deeploy.Targets.Generic.Layers.iNoNormLayer object>, 'iSoftmax': <Deeploy.Targets.Generic.Layers.SoftmaxLayer object>}, initCode='', includeList=['snrt.h', 'DeeploySnitchMath.h'])

Bases: DeploymentEngine

Methods

__init__(name: str, Mapping={'Add': <Deeploy.Targets.Generic.Layers.AddLayer object>, 'Gather': <Deeploy.Targets.Generic.Layers.GatherLayer object>, 'Gemm': <Deeploy.Targets.Generic.Layers.GEMMLayer object>, 'MatMul': <Deeploy.Targets.Generic.Layers.MatMulLayer object>, 'Pad': <Deeploy.Targets.Generic.Layers.PadLayer object>, 'RQGemm': <Deeploy.Targets.Generic.Layers.RQGEMMLayer object>, 'RQIntegerDiv': <Deeploy.Targets.Generic.Layers.RQIntegerDivLayer object>, 'RequantizedAdd': <Deeploy.Targets.Generic.Layers.AddLayer object>, 'Softmax': <Deeploy.Targets.Generic.Layers.SoftmaxLayer object>, 'Unsqueeze': <Deeploy.Targets.Generic.Layers.ReshapeLayer object>, 'iLayerNorm': <Deeploy.Targets.Generic.Layers.LayerNormLayer object>, 'iNoNorm': <Deeploy.Targets.Generic.Layers.iNoNormLayer object>, 'iSoftmax': <Deeploy.Targets.Generic.Layers.SoftmaxLayer object>}, initCode='', includeList=['snrt.h', 'DeeploySnitchMath.h']) → None

Instantiate a new engine

Parameters:

• name (str) – Name of this compute engine; must be unique per deployemnt

• Mapping (Dict[str, Union[ONNXLayer, Callable[[gs.Node], Any]]]) – Mapping between operator names and ONNXLayer implementations

• initCode (str) – Static initialization code for this engine

• includeList (List[str]) – List of header files to be included with #include directives

__init__(name[, Mapping, initCode, includeList])	Instantiate a new engine
canExecute(node)	Return whether this accelerator can execute an operator

Attributes

name	Name of this compute engine; must be unique per deployemnt
Mapping	Mapping between operator names and ONNXLayer implementations
includeList	List of header files to be included with #include directives

canExecute(node: Node) → bool

Return whether this accelerator can execute an operator

Parameters:

node (gs.Node) – Operator to be checked

Returns:

True if operator can be run on this Engine, False otherwise

Return type:

bool

name

Name of this compute engine; must be unique per deployemnt

Type:

str

Mapping

Mapping between operator names and ONNXLayer implementations

includeList

List of header files to be included with #include directives

Type:

List[str]