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]

Authors

Chao Gao

This author has not been identified. Look up 'Chao Gao' in Google

Tong Mo

This author has not been identified. Look up 'Tong Mo' in Google

Taylor Zowtuk

This author has not been identified. Look up 'Taylor Zowtuk' in Google

Tanvir Sajed

This author has not been identified. Look up 'Tanvir Sajed' in Google

Laiyuan Gong

This author has not been identified. Look up 'Laiyuan Gong' in Google

Hanxuan Chen

This author has not been identified. Look up 'Hanxuan Chen' in Google

Shangling Jui

This author has not been identified. Look up 'Shangling Jui' in Google

Wei Lu

This author has not been identified. Look up 'Wei Lu' in Google