Python MetaScheduleContext Examples

Example #1

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)

Example #2

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)

Example #3

def test_meta_schedule_integration_no_current():
    assert MetaScheduleContext.current() is None

Example #4

def test_meta_schedule_integration_current():
    env = TaskExtraction()
    with env:
        assert MetaScheduleContext.current() == env