Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning
Published in ICTAI, 2021
Recommended citation: Gao et al., Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning #
This describes our work of improving TVM’s Ansor using Bandit-based RL.