This is an automated email from the ASF dual-hosted git repository.
lmzheng pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from c1d347f [Support] Add parallel_for support to run a loop in parallel
(#6275)
add e5b793f [Ansor][AutoTVM v2.0] Phase 1: XGBoost Cost Model (#6270)
No new revisions were added by this update.
Summary of changes:
include/tvm/auto_scheduler/feature.h | 2 +-
python/tvm/auto_scheduler/__init__.py | 4 +-
python/tvm/auto_scheduler/auto_schedule.py | 22 +-
python/tvm/auto_scheduler/cost_model/__init__.py | 1 +
python/tvm/auto_scheduler/cost_model/cost_model.py | 20 +
python/tvm/auto_scheduler/cost_model/xgb_model.py | 599 +++++++++++++++++++++
python/tvm/auto_scheduler/feature.py | 44 +-
python/tvm/auto_scheduler/measure.py | 8 +-
src/auto_scheduler/cost_model.cc | 19 +
src/auto_scheduler/feature.cc | 28 +-
src/auto_scheduler/search_policy/sketch_policy.cc | 20 +-
src/auto_scheduler/search_policy/sketch_policy.h | 18 +-
.../unittest/test_auto_scheduler_cost_model.py | 51 +-
13 files changed, 779 insertions(+), 57 deletions(-)
create mode 100644 python/tvm/auto_scheduler/cost_model/xgb_model.py
