[MetaSchedule] Fix typo of compare between GlobalVar and str (apache#…

…12704) fix typo of compare between GlobalVar and str
daobook · Nov 25, 2022 · 43ede31 · 43ede31
1 parent abe913f
commit 43ede31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/meta_schedule/default_config.py b/python/tvm/meta_schedule/default_config.py
@@ -53,7 +53,7 @@ def mod(mod: Union[PrimFunc, IRModule]) -> IRModule:  # pylint: disable=redefine
         raise TypeError(f"Expected `mod` to be PrimFunc or IRModule, but gets: {mod}")
     func_names = mod.get_global_vars()
     (func_name,) = func_names
-    if len(func_names) == 1 and func_name != "main":
+    if len(func_names) == 1 and func_name.name_hint != "main":
         mod = IRModule({"main": mod[func_name]})
     return mod