def test_meta_schedule_integration_query_inside_with_scope(): mod, _, _ = get_network(name="resnet_18", input_shape=[1, 3, 224, 224]) env = TaskExtraction() with env: MetaScheduleContext.query_inside_with_scope( task_name="mock-task", mod=mod, target=Target("llvm"), dispatched=[MockModule], ) _check_mock_task(env.tasks, mod)
def test_meta_schedule_integration_query_inside_with_scope(): mod, _, _, _ = get_network( name="resnet-18", batch_size=1, layout="NHWC", dtype="float32", ) env = TaskExtraction() with env: MetaScheduleContext.query_inside_with_scope( task_name="mock-task", mod=mod, dispatched=[MockModule], ) _check_mock_task(env.tasks, mod)
def test_meta_schedule_integration_no_current(): assert MetaScheduleContext.current() is None
def test_meta_schedule_integration_current(): env = TaskExtraction() with env: assert MetaScheduleContext.current() == env