Ignore __tvm_main__ in unit test

apache · May 30, 2023 · fe44c73 · fe44c73
1 parent a42df45
commit fe44c73
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/cpp/c_codegen_test.cc b/tests/cpp/c_codegen_test.cc
@@ -121,5 +121,11 @@ TEST(CCodegen, FunctionOrder) {
   auto module = build(inputs, Target());
   Array<String> func_array = module->GetFunction("get_func_names", false)();
   std::vector<std::string> functions{func_array.begin(), func_array.end()};
+  // The entry point is handled separately from the other functions.
+  functions.erase(std::remove_if(functions.begin(), functions.end(),
+                                 [](const std::string& name) {
+                                   return name == tvm::runtime::symbol::tvm_module_main;
+                                 }),
+                  functions.end());
   EXPECT_TRUE(std::is_sorted(functions.begin(), functions.end()));
 }