f8186d8c7d
This PR introduces `Schedule.work_on`, which instructs `Schedule.get_block` to find the correct PrimFunc to retrieve from without having to specify `func_name` in every time if the PrimFunc's name is not `main`.