77
88from Deeploy .DeeployTypes import ConstantBuffer , DeploymentEngine , DeploymentPlatform , NetworkContext , NodeMapper , \
99 NodeTemplate , StructBuffer , TopologyOptimizer , TransientBuffer , VariableBuffer
10-
11- from Deeploy .Targets .Generic .Templates import AllocateTemplate as BasicAllocateTemplate
1210from Deeploy .Targets .Generic .Bindings import BasicAddBindings
13- from Deeploy .Targets .Generic .Parsers import AddParser
1411from Deeploy .Targets .Generic .Layers import AddLayer
15-
12+ from Deeploy .Targets .Generic .Parsers import AddParser
13+ from Deeploy .Targets .Generic .Templates import AllocateTemplate as BasicAllocateTemplate
1614from Deeploy .Targets .Magia .Templates import AllocateTemplate , FreeTemplate
1715
1816AddMapper = NodeMapper (AddParser (), BasicAddBindings )
1917
20- MagiaMapping = {
21- 'Add' : AddLayer ([AddMapper ])
22- }
18+ MagiaMapping = {'Add' : AddLayer ([AddMapper ])}
19+
2320
2421class MagiaVariableBuffer (VariableBuffer ):
2522
@@ -41,6 +38,7 @@ def _bufferRepresentation(self):
4138 "_memoryLevel" : memoryLevel
4239 }
4340
41+
4442class MagiaTransientBuffer (TransientBuffer ):
4543
4644 initTemplate = AllocateTemplate .magiaInitTemplate
@@ -54,11 +52,8 @@ def _bufferRepresentation(self):
5452 else :
5553 memoryLevel = None
5654
57- return {"type" : self ._type ,
58- "name" : self .name ,
59- "size" : self .size ,
60- "_memoryLevel" : memoryLevel
61- }
55+ return {"type" : self ._type , "name" : self .name , "size" : self .size , "_memoryLevel" : memoryLevel }
56+
6257
6358class MagiaConstantBuffer (ConstantBuffer ):
6459
@@ -78,6 +73,7 @@ def _bufferRepresentation(self):
7873
7974 return operatorRepresentation
8075
76+
8177class MagiaStructBuffer (StructBuffer ):
8278
8379 initTemplate = BasicAllocateTemplate .referenceStructInitTemplate
@@ -91,10 +87,7 @@ class MagiaStructBuffer(StructBuffer):
9187 ],
9288 name = "MagiaOptimizer" )
9389
94-
95- _includeList = [
96- "tile.h" , "idma.h" , "redmule.h" , "eventunit.h"
97- ]
90+ _includeList = ["tile.h" , "idma.h" , "redmule.h" , "eventunit.h" ]
9891
9992
10093class MagiaMeshEngine (DeploymentEngine ):
@@ -108,6 +101,7 @@ def __init__(self,
108101 super ().__init__ (name , Mapping , initCode , includeList )
109102 self .n_tiles = n_tiles
110103
104+
111105class MagiaPlatform (DeploymentPlatform ):
112106
113107 def __init__ (self ,
@@ -116,4 +110,4 @@ def __init__(self,
116110 constantBuffer = MagiaConstantBuffer ,
117111 structBuffer = MagiaStructBuffer ,
118112 transientBuffer = MagiaTransientBuffer ) -> None :
119- super ().__init__ (engines , variableBuffer , constantBuffer , structBuffer , transientBuffer )
113+ super ().__init__ (engines , variableBuffer , constantBuffer , structBuffer , transientBuffer )
0 commit comments