[AUTOTVM][TOPI] Use tunable templates for GPU (CUDA/OpenCL/ROCm/Mali) (#1638)
Showing
- apps/benchmark/README.md 44 additions, 19 deletionsapps/benchmark/README.md
- apps/benchmark/arm_cpu_imagenet_bench.py 55 additions, 43 deletionsapps/benchmark/arm_cpu_imagenet_bench.py
- apps/benchmark/gpu_imagenet_bench.py 45 additions, 64 deletionsapps/benchmark/gpu_imagenet_bench.py
- apps/benchmark/mobile_gpu_imagenet_bench.py 90 additions, 0 deletionsapps/benchmark/mobile_gpu_imagenet_bench.py
- apps/benchmark/util.py 11 additions, 6 deletionsapps/benchmark/util.py
- nnvm/python/nnvm/testing/__init__.py 1 addition, 0 deletionsnnvm/python/nnvm/testing/__init__.py
- nnvm/python/nnvm/testing/densenet.py 49 additions, 0 deletionsnnvm/python/nnvm/testing/densenet.py
- nnvm/python/nnvm/testing/resnet.py 2 additions, 4 deletionsnnvm/python/nnvm/testing/resnet.py
- nnvm/tests/python/frontend/mxnet/model_zoo/resnet.py 2 additions, 3 deletionsnnvm/tests/python/frontend/mxnet/model_zoo/resnet.py
- python/tvm/autotvm/measure/measure.py 25 additions, 0 deletionspython/tvm/autotvm/measure/measure.py
- python/tvm/autotvm/measure/measure_methods.py 6 additions, 6 deletionspython/tvm/autotvm/measure/measure_methods.py
- python/tvm/autotvm/record.py 1 addition, 1 deletionpython/tvm/autotvm/record.py
- python/tvm/autotvm/task/__init__.py 3 additions, 2 deletionspython/tvm/autotvm/task/__init__.py
- python/tvm/autotvm/task/dispatcher.py 28 additions, 3 deletionspython/tvm/autotvm/task/dispatcher.py
- python/tvm/autotvm/task/nnvm_integration.py 64 additions, 1 deletionpython/tvm/autotvm/task/nnvm_integration.py
- python/tvm/autotvm/task/space.py 63 additions, 6 deletionspython/tvm/autotvm/task/space.py
- python/tvm/autotvm/tophub.py 61 additions, 3 deletionspython/tvm/autotvm/tophub.py
- python/tvm/autotvm/tuner/tuner.py 4 additions, 2 deletionspython/tvm/autotvm/tuner/tuner.py
- python/tvm/autotvm/tuner/xgboost_cost_model.py 1 addition, 1 deletionpython/tvm/autotvm/tuner/xgboost_cost_model.py
- python/tvm/contrib/download.py 2 additions, 2 deletionspython/tvm/contrib/download.py
Loading
Please register or sign in to comment