Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning

Efficient execution is crucial to the successful deployment of deep learning models to real-world applications. Considerable recent effort have been devoted to computer systems for automatic discovery of efficient schedules for executing tensor programs on given hardware platforms. Built on TVM [1], Ansor [2] is the most recent and state-of-the-art framework for auto-scheduling deep neural net com...