Skip to content

Commit

Permalink
remove TODO comment on parse_mod
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed Mar 15, 2022
1 parent d4b97d0 commit 04b7ea1
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions src/meta_schedule/integration.cc
Original file line number Diff line number Diff line change
Expand Up @@ -123,12 +123,6 @@ Optional<IRModule> ApplyHistoryBestNode::Query(runtime::String task_name, IRModu
IRModule prim_mod = dispatched.value()[0];
ICHECK(HasOnlyOneFunction<tir::PrimFunc>(prim_mod)) << prim_mod;

// TODO(masahi): parse_mod below replaces the original function key with "main".
// This is necessary because, in practice, most uses of the scheduling primitive "get_block"
// assume that the function name is "main".
// If we do not have this requirement, we can remove the call to parse_mod and
// GetOnlyOneFunction* calls below and instead we can directly return sch->mod().

// Keep the original func name to be returned later.
GlobalVar gv = GetOnlyOneFunctionKey<tir::PrimFunc>(prim_mod).value();

Expand Down

0 comments on commit 04b7ea1

Please sign in to comment.