Skip to content

Commit ef032b3

Browse files
author
Matthew Brookhart
authored
Remove MemoryPlan from VM passes (apache#7361)
1 parent b8ad146 commit ef032b3

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/relay/backend/vm/compiler.cc

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -985,8 +985,11 @@ transform::Sequential MemoryOpt(tvm::Target host_target, TargetsMap targets) {
985985
// Fuse the shape functions.
986986
pass_seqs.push_back(transform::FuseOps());
987987

988-
// Perform memory planning in order to coalesce/reduce allocations.
989-
pass_seqs.push_back(transform::MemoryPlan());
988+
// TODO(mbrookhart, jroesch, masahi): this pass is very slow, and is
989+
// incomplete to provide memory resuse optimizations. Disable it until we can
990+
// rewrite it in C++ and complete it.
991+
// // Perform memory planning in order to coalesce/reduce allocations.
992+
// pass_seqs.push_back(transform::MemoryPlan());
990993

991994
// Compute away constant computation introduced by coalescing allocations.
992995
pass_seqs.push_back(transform::FoldConstant());

src/relay/backend/vm/lambda_lift.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,6 @@ class LambdaLifter : public ExprMutator {
192192
global = module_->GetGlobalVar(name);
193193
} else {
194194
// Add the lifted function to the module.
195-
std::cout << AsText(lifted_func) << std::endl;
196195
module_->Add(global, lifted_func);
197196
}
198197

0 commit comments

Comments
 (0)