We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2bd47ec commit 8b075fdCopy full SHA for 8b075fd
src/relay/backend/aot_executor_codegen.cc
@@ -638,7 +638,8 @@ class AOTExecutorCodegen : public ExprVisitor {
638
auto lowered_main = lowered_module.main_module->Lookup("main");
639
auto lowered_main_func = GetRef<Function>(lowered_main.as<FunctionNode>());
640
641
- // Post-lowering storage map for writing main func
+ // 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
643
auto new_allocator = AOTOnDemandAllocator();
644
new_allocator.Run(lowered_main_func);
645
storage_device_map_ = new_allocator.GetStorageMap();
0 commit comments