One-Shot Tuner for Deep Learning Compilers

One Shot Tuner

a neural-predictor inspired approach to auto-tuning

Docker guide

If setting the environment is difficult, try using Docker container

docker run -it --rm  jaehun/ost:v1 bash # docker running
cd /root/tvm
./main.sh                                       # start experiment
python3 get_result.py                         # get results
 

https://hub.docker.com/r/jaehun/ost