From b3b15f22bb33a166d0d73eb26153dafe90c0e79a Mon Sep 17 00:00:00 2001
From: Vincent Zhao <vincentzhaorz@gmail.com>
Date: Tue, 11 Feb 2020 21:46:13 +0000
Subject: [PATCH] Dummy func -> CSourceModule with empty code str

---
 src/relay/backend/build_module.cc | 21 +++++++--------------
 1 file changed, 7 insertions(+), 14 deletions(-)

diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc
index 4af23e2b21717..fb7250009137e 100644
--- a/src/relay/backend/build_module.cc
+++ b/src/relay/backend/build_module.cc
@@ -31,6 +31,7 @@
 #include <tvm/tir/ir_pass.h>
 #include <memory>
 
+#include "../../target/source/codegen_source_base.h"
 #include "utils.h"
 
 namespace tvm {
@@ -38,9 +39,6 @@ namespace relay {
 namespace backend {
 
 using tir::LoweredFunc;
-using tir::Stmt;
-using tir::MakeAPI;
-using tir::EvaluateNode;
 
 using TargetsMap = Map<tvm::Integer, tvm::Target>;
 using namespace tvm::relay::transform;
@@ -442,19 +440,14 @@ class RelayBuildModule : public runtime::ModuleNode {
 
     auto lowered_funcs = graph_codegen_->GetLoweredFunc();
     if (lowered_funcs.size() == 0) {
-      LOG(WARNING) << "No lowered funcs exist in the compiled module, "
-                   << "a dummy function \"__dummy__\" will be created.";
-      Stmt body = EvaluateNode::make(0);
-      Array<ObjectRef> api_args;
-      auto dummy_func = MakeAPI(body, "__dummy__", api_args, 0, false);
-      lowered_funcs.Set("llvm", Array<LoweredFunc>({dummy_func}));
+      ret_.mod = tvm::codegen::CSourceModuleCreate(std::string(), std::string());
+    } else {
+      ret_.mod = tvm::build(
+        lowered_funcs,
+        target_host_,
+        BuildConfig::Current());
     }
 
-    ret_.mod = tvm::build(
-      lowered_funcs,
-      target_host_,
-      BuildConfig::Current());
-
     Array<tvm::runtime::Module> ext_mods = graph_codegen_->GetExternalModules();
     if (!ext_mods.empty()) {
       CHECK(lowered_funcs.size() > 0 || ext_mods.size() == 1)