tqchen commented on pull request #7619: URL: https://github.com/apache/tvm/pull/7619#issuecomment-797486821
Thanks @zhuwenxi , I looked at the code, I think we can do a even smaller amount of changes here to achieve the goal: - keep most of the original logic of the max_shape, and max_arg_stack logic - Create a function "RealizeAllocaScope(Stmt body)" that will create new instance of stack values,backs up the statistics in the parent scope, visit its body, and create the alloca at its current scope(similaer to Build function) - Call RealizeAllocaScope in a ParallelFor(so we make sure alloca is only lifted there) and root ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
