Skip to content

Commit 04b7ea1

Browse files
committed
remove TODO comment on parse_mod
1 parent d4b97d0 commit 04b7ea1

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/meta_schedule/integration.cc

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -123,12 +123,6 @@ Optional<IRModule> ApplyHistoryBestNode::Query(runtime::String task_name, IRModu
123123
IRModule prim_mod = dispatched.value()[0];
124124
ICHECK(HasOnlyOneFunction<tir::PrimFunc>(prim_mod)) << prim_mod;
125125

126-
// TODO(masahi): parse_mod below replaces the original function key with "main".
127-
// This is necessary because, in practice, most uses of the scheduling primitive "get_block"
128-
// assume that the function name is "main".
129-
// If we do not have this requirement, we can remove the call to parse_mod and
130-
// GetOnlyOneFunction* calls below and instead we can directly return sch->mod().
131-
132126
// Keep the original func name to be returned later.
133127
GlobalVar gv = GetOnlyOneFunctionKey<tir::PrimFunc>(prim_mod).value();
134128

0 commit comments

Comments
 (0)