Adding manual scratchpad management code in custom TIR pass

[XYX]

Hello, is your problem solved now? I also encountered a similar problem.