Compiler Auto-Vectorization with Imitation Learning

1 code implementation • NeurIPS 2019 • Charith Mendis, Cambridge Yang, Yewen Pu, Dr.Saman Amarasinghe, Michael Carbin

We show that the learnt policy produces a vectorization scheme which is better than industry standard compiler heuristics both in terms of static measures and runtime performance.

Imitation Learning