Skip to content

Commit 83c56f3

Browse files
Luka Macanlukamac
authored andcommitted
Add missing variable replacement update pass
1 parent 877acbc commit 83c56f3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

Deeploy/Targets/Snitch/Bindings.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@
4141
from Deeploy.Targets.Snitch.Templates.FloatSoftmaxTemplate import FloatSoftmax_Template
4242
from Deeploy.Targets.Snitch.Templates.GemmTemplate import SnitchGemm_Template
4343
from Deeploy.Targets.Snitch.Templates.RqGemmTemplate import SnitchRqGemm_Template
44-
from Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement import TilingVariableReplacement
44+
from Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement import TilingVariableReplacement, \
45+
TilingVariableReplacementUpdate
4546

4647
TilingCallClosure = partial(ClosureGeneration, closureSuffix = "_tiling_closure")
4748
MemoryAwareFunctionCallClosure = partial(MemoryAwareClosureGeneration,
@@ -61,7 +62,8 @@
6162
TilingVariableReplacement("L1"),
6263
TilingCallClosure(writeback = False),
6364
SnitchSynchCoresPass(),
64-
SnitchClusterTiling("L2", "L1", SnitchDma()), # TODO
65+
SnitchClusterTiling("L2", "L1", SnitchDma()),
66+
TilingVariableReplacementUpdate("L1"),
6567
ArgumentStructGeneration(),
6668
MemoryManagementGeneration("L1"),
6769
MemoryAwareFunctionCallClosure(writeback = False, generateStruct = True),

0 commit comments

Comments
 (0)