Looplets: A Language For Structured Coiteration

Abstract

Real world arrays often contain underlying structure, such as sparsity, runs of repeated values, or symmetry. Specializing for structure yields significant speedups. But automatically generating efficient code for structured data is challenging, especially when arrays with different structure interact. We show how to abstract over array structures so that the compiler can generate code to coiterate over any combination of them. Our technique enables new array formats (such as 1DVBL for irregular clustered sparsity), new iteration strategies (such as galloping intersections), and new operations over structured data (such as concatenation or convolution).

Article

BibTeX

@article{willow2023,
  title={Looplets: A Language For Structured Coiteration},
  author={Willow Ahrens and Daniel Donenfeld and Fredrik Kjolstad and Saman Amarasinghe},
  journal={International Symposium on Code Generation and Optimization},
  year={2023},
  month={February}
}