|
41 | 41 | from Deeploy.Targets.Snitch.Templates.FloatSoftmaxTemplate import FloatSoftmax_Template |
42 | 42 | from Deeploy.Targets.Snitch.Templates.GemmTemplate import SnitchGemm_Template |
43 | 43 | from Deeploy.Targets.Snitch.Templates.RqGemmTemplate import SnitchRqGemm_Template |
| 44 | +<<<<<<< Updated upstream |
44 | 45 | from Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement import TilingVariableReplacement |
| 46 | +======= |
| 47 | +from Deeploy.TilingExtension.AsyncDma import AsyncDma |
| 48 | +from Deeploy.TilingExtension.CodeTransformationPasses.TilingVariableReplacement import TilingVariableReplacement, TilingVariableReplacementUpdate |
| 49 | +>>>>>>> Stashed changes |
45 | 50 |
|
46 | 51 | TilingCallClosure = partial(ClosureGeneration, closureSuffix = "_tiling_closure") |
47 | 52 | MemoryAwareFunctionCallClosure = partial(MemoryAwareClosureGeneration, |
|
61 | 66 | TilingVariableReplacement("L1"), |
62 | 67 | TilingCallClosure(writeback = False), |
63 | 68 | SnitchSynchCoresPass(), |
| 69 | +<<<<<<< Updated upstream |
64 | 70 | SnitchClusterTiling("L2", "L1", SnitchDma()), # TODO |
| 71 | +======= |
| 72 | + TilingVariableReplacementUpdate("L1"), |
| 73 | + SnitchClusterTiling("L2?", "L1", AsyncDma({})), |
| 74 | +>>>>>>> Stashed changes |
65 | 75 | ArgumentStructGeneration(), |
66 | 76 | MemoryManagementGeneration("L1"), |
67 | 77 | MemoryAwareFunctionCallClosure(writeback = False, generateStruct = True), |
|
0 commit comments