Skip to content

Commit 74a02d9

Browse files
committed
Fix small mistake
1 parent 1e5c9e0 commit 74a02d9

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

Deeploy/DeeployTypes.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1972,7 +1972,7 @@ def discardCurrentMapper(self):
19721972
"""Discard the current Mapper
19731973
19741974
"""
1975-
self.dicardedMappers.add(self.mapper)
1975+
self.discardedMappers.add(self.mapper)
19761976
self.mapper = None
19771977

19781978
def resetDiscardedMappers(self):
@@ -2124,7 +2124,7 @@ def bind(self, ctxt: NetworkContext) -> Tuple[NetworkContext, bool]:
21242124
self.mapper.parser.operatorRepresentation['nodeOps'] = int(self.computeOps())
21252125
return newCtxt, True
21262126

2127-
self.discardedMappers.append(self.mapper)
2127+
self.discardedMappers.add(self.mapper)
21282128
return ctxt, False
21292129

21302130
def codeTransform(self, ctxt: NetworkContext, verbose: CodeGenVerbosity = _NoVerbosity) -> NetworkContext:
@@ -3175,7 +3175,7 @@ def importDeeployState(self, folderPath: str, fileName: str):
31753175
31763176
"""
31773177
self.graph = NetworkDeployer._importONNXGraph(folderPath, f"{fileName}")
3178-
self.ctxt = NetworkContext.importNetworkCtxt(folderPath, f"{fileName}")
3178+
self.ctxt = NetworkContext.importNetworkContext(folderPath, f"{fileName}")
31793179

31803180

31813181
class NetworkDeployer(NetworkContainer):

DeeployTest/testMVP.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def setupDeployer(graph: gs.Graph, memoryHierarchy: MemoryHierarchy, defaultTarg
143143
deployer.tiler.memoryAllocStrategy = args.memAllocStrategy
144144
deployer.tiler.searchStrategy = args.searchStrategy
145145

146-
return deployer
146+
return deployer, signProp
147147

148148

149149
if __name__ == '__main__':
@@ -263,14 +263,13 @@ def setupDeployer(graph: gs.Graph, memoryHierarchy: MemoryHierarchy, defaultTarg
263263
memoryHierarchy = MemoryHierarchy(memoryLevels)
264264
memoryHierarchy.setDefaultMemoryLevel(args.defaultMemLevel)
265265

266-
deployer = setupDeployer(graph,
267-
memoryHierarchy,
268-
defaultTargetMemoryLevel = L1,
269-
defaultIoMemoryLevel = memoryHierarchy.memoryLevels[args.defaultMemLevel],
270-
verbose = verbosityCfg)
266+
deployer, signProp = setupDeployer(graph,
267+
memoryHierarchy,
268+
defaultTargetMemoryLevel = L1,
269+
defaultIoMemoryLevel = memoryHierarchy.memoryLevels[args.defaultMemLevel],
270+
verbose = verbosityCfg)
271271

272272
platform = deployer.Platform
273-
signProp = False
274273

275274
for index, num in enumerate(test_inputs):
276275
_type, offset = inferTypeAndOffset(num, signProp)

0 commit comments

Comments
 (0)