[AUTOTVM] Allow fallback for template & Fix bugs in tuners (#1615)
* support fallback & fix bugs in tuners & clean topi test * update task extraction * update task extraction * fix arm tutorial * Update tune_nnvm_arm.py
Showing
- nnvm/python/nnvm/compiler/build_module.py 3 additions, 2 deletionsnnvm/python/nnvm/compiler/build_module.py
- nnvm/tests/python/compiler/test_autotvm_task_extraction.py 63 additions, 0 deletionsnnvm/tests/python/compiler/test_autotvm_task_extraction.py
- python/tvm/autotvm/__init__.py 2 additions, 1 deletionpython/tvm/autotvm/__init__.py
- python/tvm/autotvm/measure/measure.py 3 additions, 2 deletionspython/tvm/autotvm/measure/measure.py
- python/tvm/autotvm/measure/measure_methods.py 25 additions, 19 deletionspython/tvm/autotvm/measure/measure_methods.py
- python/tvm/autotvm/task/__init__.py 1 addition, 1 deletionpython/tvm/autotvm/task/__init__.py
- python/tvm/autotvm/task/dispatcher.py 83 additions, 34 deletionspython/tvm/autotvm/task/dispatcher.py
- python/tvm/autotvm/task/nnvm_integration.py 87 additions, 30 deletionspython/tvm/autotvm/task/nnvm_integration.py
- python/tvm/autotvm/task/space.py 54 additions, 2 deletionspython/tvm/autotvm/task/space.py
- python/tvm/autotvm/task/task.py 1 addition, 1 deletionpython/tvm/autotvm/task/task.py
- python/tvm/autotvm/tophub.py 2 additions, 5 deletionspython/tvm/autotvm/tophub.py
- python/tvm/autotvm/tuner/ga_tuner.py 3 additions, 7 deletionspython/tvm/autotvm/tuner/ga_tuner.py
- python/tvm/autotvm/tuner/model_based_tuner.py 16 additions, 17 deletionspython/tvm/autotvm/tuner/model_based_tuner.py
- python/tvm/autotvm/tuner/sa_model_optimizer.py 1 addition, 1 deletionpython/tvm/autotvm/tuner/sa_model_optimizer.py
- python/tvm/autotvm/tuner/tuner.py 10 additions, 3 deletionspython/tvm/autotvm/tuner/tuner.py
- python/tvm/autotvm/tuner/xgboost_cost_model.py 81 additions, 38 deletionspython/tvm/autotvm/tuner/xgboost_cost_model.py
- python/tvm/autotvm/tuner/xgboost_tuner.py 13 additions, 4 deletionspython/tvm/autotvm/tuner/xgboost_tuner.py
- python/tvm/exec/tophub.py 5 additions, 4 deletionspython/tvm/exec/tophub.py
- python/tvm/target.py 1 addition, 0 deletionspython/tvm/target.py
- tests/python/unittest/test_autotvm_dispatch_context.py 29 additions, 15 deletionstests/python/unittest/test_autotvm_dispatch_context.py
Loading
Please register or sign in to comment