ProTuner: Tuning Programs with Monte Carlo Tree Search

作者: Ameer Haj-Ali , Hasan Genc , Qijing Huang , William Moses , John Wawrzynek

DOI:

关键词: Computer engineeringImage processingArtificial intelligenceMonte Carlo tree searchScheduling (computing)Beam searchComputer scienceDeep learning

摘要: … We explore applying the Monte Carlo Tree Search (MCTS) algorithm in a notoriously difficult task: tuning programs for high-performance deep learning and image processing. We build …

参考文章(46)
Csaba Szepesvári, Jan Willemson, Levente Kocsis, Improved Monte-Carlo Search ,(2006)
Guillaume M. J. -B. Chaslot, Mark H. M. Winands, H. Jaap van den Herik, Parallel Monte-Carlo Tree Search Computers and Games. pp. 60- 71 ,(2008) , 10.1007/978-3-540-87608-3_6
Qijing Huang, Ruolong Lian, Andrew Canis, Jongsok Choi, Ryan Xi, Nazanin Calagar, Stephen Brown, Jason Anderson, None, The Effect of Compiler Optimizations on High-Level Synthesis-Generated Hardware ACM Transactions on Reconfigurable Technology and Systems. ,vol. 8, pp. 14- ,(2015) , 10.1145/2629547
Qijing Huang, Ruolong Lian, Andrew Canis, Jongsok Choi, Ryan Xi, Stephen Brown, Jason Anderson, None, The Effect of Compiler Optimizations on High-Level Synthesis for FPGAs field-programmable custom computing machines. pp. 89- 96 ,(2013) , 10.1109/FCCM.2013.50
Vasileios Porpodas, Timothy M. Jones, Alberto Magni, PSLP: padded SLP automatic vectorization symposium on code generation and optimization. pp. 190- 201 ,(2015) , 10.5555/2738600.2738625
Daniel S. McFarlin, Volodymyr Arbatov, Franz Franchetti, Markus Püschel, Automatic SIMD vectorization of fast fourier transforms for the larrabee and AVX instruction sets Proceedings of the international conference on Supercomputing - ICS '11. pp. 265- 274 ,(2011) , 10.1145/1995896.1995938
Uday Bondhugula, Albert Hartono, J. Ramanujam, P. Sadayappan, A practical automatic polyhedral parallelizer and locality optimizer Proceedings of the 2008 ACM SIGPLAN conference on Programming language design and implementation - PLDI '08. ,vol. 43, pp. 101- 113 ,(2008) , 10.1145/1375581.1375595
Jonathan Ragan-Kelley, Connelly Barnes, Andrew Adams, Sylvain Paris, Frédo Durand, Saman Amarasinghe, Halide: a language and compiler for optimizing parallelism, locality, and recomputation in image processing pipelines programming language design and implementation. ,vol. 48, pp. 519- 530 ,(2013) , 10.1145/2491956.2462176
Kevin Stock, Louis-Noël Pouchet, P. Sadayappan, Using machine learning to improve automatic vectorization high performance embedded architectures and compilers. ,vol. 8, pp. 50- ,(2012) , 10.1145/2086696.2086729
Ravi Teja Mullapudi, Vinay Vasista, Uday Bondhugula, PolyMage: Automatic Optimization for Image Processing Pipelines architectural support for programming languages and operating systems. ,vol. 50, pp. 429- 443 ,(2015) , 10.1145/2694344.2694364