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]

@inproceedings{GaoMZSGCJL21,
  title = {Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning},
  author = {Chao Gao and Tong Mo and Taylor Zowtuk and Tanvir Sajed and Laiyuan Gong and Hanxuan Chen and Shangling Jui and Wei Lu},
  year = {2021},
  doi = {10.1109/ICTAI52525.2021.00045},
  url = {https://doi.org/10.1109/ICTAI52525.2021.00045},
  researchr = {https://researchr.org/publication/GaoMZSGCJL21},
  cites = {0},
  citedby = {0},
  pages = {273-278},
  booktitle = {33rd IEEE International Conference on Tools with Artificial Intelligence, ICTAI 2021, Washington, DC, USA, November 1-3, 2021},
  publisher = {IEEE},
  isbn = {978-1-6654-0898-1},
}