Skip to content

Commit

Permalink
[MetaSchedule] Update Tuning Interfaces. (apache#10367)
Browse files Browse the repository at this point in the history
This PR is further improvement of the meta schedule project (apache#8473).

Co-authored-by: Junru Shao <<junrushao1994@gmail.com>>
Co-authored-by: Bohan Hou <<32121147+spectrometerHBH@users.noreply.github.com>>
Co-authored-by: Ruihang Lai <<lairuihangdongdong@qq.com>>
Co-authored-by: Hongyi Jin <<3231950289@qq.com>>
Co-authored-by: Wuwei Lin <<wuwei@apache.org>>
Co-authored-by: Siyuan Feng <<Hzfengsy@sjtu.edu.cn>>
  • Loading branch information
7 people authored and pfk-beta committed Apr 11, 2022
1 parent 4f2396c commit 7cf1ef4
Show file tree
Hide file tree
Showing 4 changed files with 162 additions and 81 deletions.
2 changes: 2 additions & 0 deletions include/tvm/meta_schedule/tune_context.h
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,8 @@ class TuneContextNode : public runtime::Object {
v->Visit("rand_state", &rand_state);
v->Visit("num_threads", &num_threads);
v->Visit("is_stopped", &is_stopped);
v->Visit("builder_results", &builder_results);
v->Visit("runner_futures", &runner_futures);
v->Visit("measure_candidates", &measure_candidates);
}

Expand Down
6 changes: 4 additions & 2 deletions python/tvm/meta_schedule/integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,15 @@

from tvm._ffi import register_object
from tvm.ir import IRModule, transform
from tvm.relay import Any, Function as RelayFunc, vm
from tvm.relay import Any
from tvm.relay import Function as RelayFunc
from tvm.relay import vm
from tvm.runtime import NDArray, Object
from tvm.target import Target
from tvm.tir import PrimFunc

from .database import Database
from . import _ffi_api
from .database import Database


@register_object("meta_schedule.ExtractedTask")
Expand Down
Loading

0 comments on commit 7cf1ef4

Please sign in to comment.