projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: db6e0c1) | patch
[Ansor][AutoTVM v2.0] Phase 2: Basic CPU Sketch Search Policy (#6184)
authorChenfan <chengfan.jcf@alibaba-inc.com>
Tue, 11 Aug 2020 20:53:00 +0000 (04:53 +0800)
committerGitHub <noreply@github.com>
Tue, 11 Aug 2020 20:53:00 +0000 (13:53 -0700)
commit75b8318a5011eca95b9e6e457fde07cf94c8f79f
treed33f10d4018f4ab2b21d171dd44fd6965dd20b72tree | snapshot
parentdb6e0c1804961d701fe39f3dce453b147113ed9bcommit | diff
[Ansor][AutoTVM v2.0] Phase 2: Basic CPU Sketch Search Policy (#6184)

* Init commit to pass the compile

* First commit to pass the test

* Update

* Add UTs for sketch generation

* Update

* Add ASF to new UT file.

* Update rule for winograd

* Update

* File renamed

* Lint fix
26 files changed:
include/tvm/auto_scheduler/auto_schedule.h diff | blob | history
include/tvm/auto_scheduler/compute_dag.h diff | blob | history
include/tvm/auto_scheduler/cost_model.h diff | blob | history
include/tvm/auto_scheduler/search_policy.h diff | blob | history
include/tvm/auto_scheduler/transform_step.h diff | blob | history
python/tvm/auto_scheduler/__init__.py diff | blob | history
python/tvm/auto_scheduler/auto_schedule.py diff | blob | history
src/auto_scheduler/auto_schedule.cc diff | blob | history
src/auto_scheduler/compute_dag.cc diff | blob | history
src/auto_scheduler/cost_model.cc diff | blob | history
src/auto_scheduler/loop_state.cc diff | blob | history
src/auto_scheduler/search_policy/empty_policy.cc diff | blob | history
src/auto_scheduler/search_policy/empty_policy.h diff | blob | history
src/auto_scheduler/search_policy/search_policy.cc diff | blob | history
src/auto_scheduler/search_policy/sketch_policy.cc [new file with mode: 0644] blob
src/auto_scheduler/search_policy/sketch_policy.h [new file with mode: 0644] blob
src/auto_scheduler/search_policy/sketch_policy_rules.cc [new file with mode: 0644] blob
src/auto_scheduler/search_policy/sketch_policy_rules.h [new file with mode: 0644] blob
src/auto_scheduler/search_policy/utils.cc [new file with mode: 0644] blob
src/auto_scheduler/search_policy/utils.h [new file with mode: 0644] blob
src/auto_scheduler/utils.h diff | blob | history
tests/cpp/auto_scheduler_test.cc diff | blob | history
tests/python/unittest/test_auto_scheduler_common.py diff | blob | history
tests/python/unittest/test_auto_scheduler_loop_state.py diff | blob | history
tests/python/unittest/test_auto_scheduler_search_policy.py diff | blob | history
tests/python/unittest/test_auto_scheduler_sketch_generation.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;