projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 5721387) | patch
[Ansor][AutoTVM v2.0] Phase 1: The base class for cost models (#6187)
authorLianmin Zheng <lianminzheng@gmail.com>
Thu, 6 Aug 2020 10:46:12 +0000 (03:46 -0700)
committerGitHub <noreply@github.com>
Thu, 6 Aug 2020 10:46:12 +0000 (18:46 +0800)
commit7ef89ad2dec184a033ed135dbef9fa599f55808b
tree13dfe1280471132600d64857a12d983cd63cee97tree | snapshot
parent57213879e6ccaa4c0e2ba08b0dca075b623a8742commit | diff
[Ansor][AutoTVM v2.0] Phase 1: The base class for cost models (#6187)

* add the base class for cost models

* address comments

* Update tests/python/unittest/test_auto_scheduler_cost_model.py

Disable test if user doesn't have llvm

Co-authored-by: Zhao Wu <zhaowu@apache.org>
include/tvm/auto_scheduler/cost_model.h [new file with mode: 0644] blob
python/tvm/auto_scheduler/__init__.py diff | blob | history
python/tvm/auto_scheduler/cost_model/__init__.py [new file with mode: 0644] blob
python/tvm/auto_scheduler/cost_model/cost_model.py [new file with mode: 0644] blob
src/auto_scheduler/cost_model.cc [new file with mode: 0644] blob
tests/python/unittest/test_auto_scheduler_cost_model.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;