Don't use Update : IRModule because it is broken

apache · Aug 20, 2021 · 9015163 · 9015163
1 parent 120b14a
commit 9015163
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/relay/backend/te_compiler.cc b/src/relay/backend/te_compiler.cc
@@ -875,7 +875,9 @@ IRModule LoweredModuleToIRModule(LoweredModule mod) {
   IRModule unified_module;
 
   // Copy the main module and its typedefs
-  unified_module->Update(mod.main_module);
+  for (const auto& kv : mod.main_module->functions) {
+    unified_module->Add(kv.first, kv.second);
+  }
   for (const auto& kv : mod.main_module->type_definitions) {
     unified_module->AddTypeDef(kv.first, kv.second);
   }