Skip to content

Commit 8b075fd

Browse files
committed
Clarify reasoning behind replanning memory later
1 parent 2bd47ec commit 8b075fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/relay/backend/aot_executor_codegen.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -638,7 +638,8 @@ class AOTExecutorCodegen : public ExprVisitor {
638638
auto lowered_main = lowered_module.main_module->Lookup("main");
639639
auto lowered_main_func = GetRef<Function>(lowered_main.as<FunctionNode>());
640640

641-
// Post-lowering storage map for writing main func
641+
// Post-lowering storage map for writing main func - this should be the same map as previously
642+
// created, just referencing the new expressions created from lowering
642643
auto new_allocator = AOTOnDemandAllocator();
643644
new_allocator.Run(lowered_main_func);
644645
storage_device_map_ = new_allocator.GetStorageMap();

0 commit comments

Comments
 (0)