review.tizen.org Git - platform/upstream/tvm.git/commit

projects / platform / upstream / tvm.git / commit

summary | shortlog | log | commit | commitdiff | tree
(parent: c1d347f) | patch

author	Lianmin Zheng <lianminzheng@gmail.com>
	Wed, 19 Aug 2020 00:51:10 +0000 (17:51 -0700)
committer	GitHub <noreply@github.com>
	Wed, 19 Aug 2020 00:51:10 +0000 (17:51 -0700)
commit	e5b793f39fd5b4f84b0aedf06aa376ebe45cf2bc
tree	50a1e22308e8621b8f50d22f40247c2ffa16a810	tree \| snapshot
parent	c1d347fbd56c2cd9622afeb706eff85ac1525c3f	commit \| diff

[Ansor][AutoTVM v2.0] Phase 1: XGBoost Cost Model (#6270)

* port xgb cost model

* add xgboost cost model

* fix lint

* address comments

* address comments

* Fix

13 files changed:

include/tvm/auto_scheduler/feature.h		diff \| blob \| history
python/tvm/auto_scheduler/__init__.py		diff \| blob \| history
python/tvm/auto_scheduler/auto_schedule.py		diff \| blob \| history
python/tvm/auto_scheduler/cost_model/__init__.py		diff \| blob \| history
python/tvm/auto_scheduler/cost_model/cost_model.py		diff \| blob \| history
python/tvm/auto_scheduler/cost_model/xgb_model.py	[new file with mode: 0644]	blob
python/tvm/auto_scheduler/feature.py		diff \| blob \| history
python/tvm/auto_scheduler/measure.py		diff \| blob \| history
src/auto_scheduler/cost_model.cc		diff \| blob \| history
src/auto_scheduler/feature.cc		diff \| blob \| history
src/auto_scheduler/search_policy/sketch_policy.cc		diff \| blob \| history
src/auto_scheduler/search_policy/sketch_policy.h		diff \| blob \| history
tests/python/unittest/test_auto_scheduler_cost_model.py		diff \| blob \| history

Domain: Machine Learning / ML Framework;