projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 40450af) | patch
[Ansor][AutoTVM v2.0] Phase 1: feature extraction for cost models (#6190)
authorLianmin Zheng <lianminzheng@gmail.com>
Wed, 12 Aug 2020 21:00:49 +0000 (14:00 -0700)
committerGitHub <noreply@github.com>
Wed, 12 Aug 2020 21:00:49 +0000 (14:00 -0700)
commit356588996a8862677df1f22030313df5f2cbf774
tree58bf6328feac99bb4b46492266da667127db5254tree | snapshot
parent40450af983ab74385630620210ee4c670a0963dbcommit | diff
[Ansor][AutoTVM v2.0] Phase 1: feature extraction for cost models (#6190)

* [AutoScheduler] add feature extraction

* fix lint

* fix gpu test

* address comments

* improve flop estimation

* rebase

* refactor with group

* fix

* Apply suggestions from code review
include/tvm/auto_scheduler/feature.h [new file with mode: 0644] blob
python/tvm/auto_scheduler/__init__.py diff | blob | history
python/tvm/auto_scheduler/feature.py [new file with mode: 0644] blob
src/auto_scheduler/compute_dag.cc diff | blob | history
src/auto_scheduler/feature.cc [new file with mode: 0644] blob
tests/python/unittest/test_auto_scheduler_compute_dag.py diff | blob | history
tests/python/unittest/test_auto_scheduler_feature.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;