File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 4141from Deeploy .Targets .Snitch .Templates .FloatSoftmaxTemplate import FloatSoftmax_Template
4242from Deeploy .Targets .Snitch .Templates .GemmTemplate import SnitchGemm_Template
4343from 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
4647TilingCallClosure = partial (ClosureGeneration , closureSuffix = "_tiling_closure" )
4748MemoryAwareFunctionCallClosure = partial (MemoryAwareClosureGeneration ,
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 ),
You can’t perform that action at this time.
0 commit comments