[AutoTVM]Core functionality for Graph tuner (#2184)
authorYao Wang <kevinthesunwy@gmail.com>
Wed, 29 May 2019 23:36:05 +0000 (16:36 -0700)
committerYizhi Liu <liuyizhi@apache.org>
Wed, 29 May 2019 23:36:05 +0000 (16:36 -0700)
commitc8a0f524d92937dd14bc400a95dee9efb1acb2b0
tree3b23cc93dda4dce7dbb35e56f2418b8812997979
parent4767554c580f5aa0521727629e51f04f5315f025
[AutoTVM]Core functionality for Graph tuner (#2184)

* Add graph tuning

* Add tests

* Fix tests

* Fix pylint

* Small fix for docstring

* Minor fix

* Support fetching workload from relay expr

* Simplify benchmark layout transformation

* Add relay support

* Fix infer layout func name

* Refactor internal data representation

* Fix issues

* Add PBQP solver

* Fix layout transform check

* Add PBQPTuner test

* Fix lint

* Update tutorial

* Fix tutorial

* Fix lint

* Add relay test

* Remove nnvm since nnvm graph can be converted to relay function

* Modify benchmark layout wrt new layout_transform api

* Fix lint

* Update docstring for DP tuner

* Refactor traverse graph

* Support graph tuning for multiple target operators

* Fix fetching workloads

* Add x86 depthwise_conv2d infer_layout

* Fix x86 depthwise_conv2d autotvm

* Fix PBQP tuner

* Fix DP tuner

* Generate dummy layout transform record

* Update tutorial

* Modify layout records name

* Add ASF header

* Add ASF header for testing files

* Fix test

* Fix topi fetching

* Some refactors

* Fix lint

* Fix tutorial

* Rename test files

* Fix doc typo

* Add test case note link
18 files changed:
python/tvm/autotvm/graph_tuner/__init__.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/_base.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/base_graph_tuner.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/dynamic_programming_stage.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/dynamic_programming_tuner.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/pbqp_tuner.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/utils/__init__.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/utils/traverse_graph.py [new file with mode: 0644]
python/tvm/autotvm/graph_tuner/utils/utils.py [new file with mode: 0644]
python/tvm/autotvm/task/__init__.py
python/tvm/autotvm/task/topi_integration.py
tests/python/unittest/test_graph_tuner_core.py [new file with mode: 0644]
tests/python/unittest/test_graph_tuner_utils.py [new file with mode: 0644]
topi/python/topi/nn/conv2d.py
topi/python/topi/nn/depthwise_conv2d.py
topi/python/topi/x86/conv2d.py
topi/python/topi/x86/depthwise_conv2d.py
tutorials/autotvm/tune_relay_x86.py