tqchen commented on code in PR #14511:
URL: https://github.com/apache/tvm/pull/14511#discussion_r1158829644
##########
src/relax/transform/static_plan_block_memory.cc:
##########
@@ -476,11 +521,15 @@ class StorageAllocatorInit : public
StorageAllocatorBaseVisitor {
token2block_.erase(token_to_discard.get());
}
+ /*! \brief The arithmetic analyzer. */
+ arith::Analyzer ana;
/*!
* \brief The context IRModule, used for checking if a callee function is
* a PrimFunc inside the IRModule.
*/
const IRModule& ctx_mod_;
+ /*! \brief The mapping from TIR variables to their respective upper bound
values. */
+ std::unordered_map<const tir::VarNode*, IntImm> var_upper_bound_;
Review Comment:
prefer `unordered_map<tir::Var, IntImm, ObjectPtrHash, ObjectPtrEqual>`,
mainly to avoid var get de-allocated and re-allocated, does not matter in here
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
