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

Chao Gao, Tong Mo, Taylor Zowtuk, Tanvir Sajed, Laiyuan Gong, Hanxuan Chen, Shangling Jui, Wei Lu. Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning. In 33rd IEEE International Conference on Tools with Artificial Intelligence, ICTAI 2021, Washington, DC, USA, November 1-3, 2021. pages 273-278, IEEE, 2021. [doi]

Abstract

Abstract is missing.