Fix tests - 2

Change-Id: I1418030e5c25a9b8d97cc741210d2866f010b72a
apache · May 12, 2021 · 71dd730 · 71dd730
1 parent e05eb8a
commit 71dd730
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc
@@ -120,7 +120,9 @@ struct AOTCodegen : ExecutorCodegen {
  mod = (*pf)();
  }
 
- void Codegen(const Function& func, String mod_name) { CallFunc("codegen", func, mod_name); }
+ void Codegen(const Function& func, String mod_name) override {
+ CallFunc("codegen", func, mod_name);
+ }
 
  void UpdateOutput(BuildOutput* ret) override { ret->graph_json = ""; }
 
@@ -138,7 +140,7 @@ struct GraphCodegen : ExecutorCodegen {
  }
  void UpdateOutput(BuildOutput* ret) override { ret->graph_json = GetGraphJSON(); }
 
- void Codegen(const Function& func, String mod_name) { CallFunc("codegen", func); }
+ void Codegen(const Function& func, String mod_name) override { CallFunc("codegen", func); }
 
  std::string GetGraphJSON() { return CallFunc<std::string>("get_graph_json", nullptr); }