From 96bdde62e88fb1e43009e95f627070c4441d6948 Mon Sep 17 00:00:00 2001
From: Wuwei Lin <wuwei@apache.org>
Date: Mon, 9 Sep 2024 14:44:24 -0700
Subject: [PATCH] [Relax] Fix BYOC removing existing ext mods

---
 src/relax/transform/run_codegen.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/relax/transform/run_codegen.cc b/src/relax/transform/run_codegen.cc
index fe0e73d99e99..af9ed2fffce2 100644
--- a/src/relax/transform/run_codegen.cc
+++ b/src/relax/transform/run_codegen.cc
@@ -79,6 +79,10 @@ class CodeGenRunner : ExprMutator {
     auto out_mod = builder_->GetContextIRModule();
 
     if (ext_mods.size()) {
+      if (auto opt_old_ext_mods = mod->GetAttr<Array<runtime::Module>>(tvm::attr::kExternalMods)) {
+        auto old_ext_mods = opt_old_ext_mods.value();
+        ext_mods.insert(ext_mods.begin(), old_ext_mods.begin(), old_ext_mods.end());
+      }
       out_mod = WithAttr(out_mod, tvm::attr::kExternalMods, std::move(ext_mods));
     }