1717
1818AddMapper = NodeMapper (AddParser (), BasicAddBindings )
1919
20- MagiaMapping = {
21- 'Add' : AddLayer ([AddMapper ])
22- }
20+ MagiaMapping = {'Add' : AddLayer ([AddMapper ])}
21+
2322
2423class MagiaVariableBuffer (VariableBuffer ):
2524
@@ -41,6 +40,7 @@ def _bufferRepresentation(self):
4140 "_memoryLevel" : memoryLevel
4241 }
4342
43+
4444class MagiaTransientBuffer (TransientBuffer ):
4545
4646 initTemplate = AllocateTemplate .magiaInitTemplate
@@ -54,11 +54,8 @@ def _bufferRepresentation(self):
5454 else :
5555 memoryLevel = None
5656
57- return {"type" : self ._type ,
58- "name" : self .name ,
59- "size" : self .size ,
60- "_memoryLevel" : memoryLevel
61- }
57+ return {"type" : self ._type , "name" : self .name , "size" : self .size , "_memoryLevel" : memoryLevel }
58+
6259
6360class MagiaConstantBuffer (ConstantBuffer ):
6461
@@ -78,6 +75,7 @@ def _bufferRepresentation(self):
7875
7976 return operatorRepresentation
8077
78+
8179class MagiaStructBuffer (StructBuffer ):
8280
8381 initTemplate = BasicAllocateTemplate .referenceStructInitTemplate
@@ -91,10 +89,7 @@ class MagiaStructBuffer(StructBuffer):
9189 ],
9290 name = "MagiaOptimizer" )
9391
94-
95- _includeList = [
96- "tile.h" , "idma.h" , "redmule.h" , "eventunit.h"
97- ]
92+ _includeList = ["tile.h" , "idma.h" , "redmule.h" , "eventunit.h" ]
9893
9994
10095class MagiaMeshEngine (DeploymentEngine ):
@@ -108,6 +103,7 @@ def __init__(self,
108103 super ().__init__ (name , Mapping , initCode , includeList )
109104 self .n_tiles = n_tiles
110105
106+
111107class MagiaPlatform (DeploymentPlatform ):
112108
113109 def __init__ (self ,
@@ -116,4 +112,4 @@ def __init__(self,
116112 constantBuffer = MagiaConstantBuffer ,
117113 structBuffer = MagiaStructBuffer ,
118114 transientBuffer = MagiaTransientBuffer ) -> None :
119- super ().__init__ (engines , variableBuffer , constantBuffer , structBuffer , transientBuffer )
115+ super ().__init__ (engines , variableBuffer , constantBuffer , structBuffer , transientBuffer )
0 commit comments