124 citations · 315 across the 19 of their papers we have counts for
3 papers · 1 filter
An Asymptotic Cost Model for Autoscheduling Sparse Tensor Programs
Willow Ahrens, Fredrik Kjolstad, Saman Amarasinghe
While loop reordering and fusion can make big impacts on the constant-factor performance of dense tensor programs, the effects on sparse tensor programs are asymptotic, often leadi…
An Attempt to Generate Code for Symmetric Tensor Computations
Jessica Shi, Stephen Chou, Fredrik Kjolstad +1
This document describes an attempt to develop a compiler-based approach for computations with symmetric tensors. Given a computation and the symmetries of its input tensors, we der…
Compiling Halide Programs to Push-Memory Accelerators
Qiaoyi Liu, Dillon Huff, Jeff Setter +8
Image processing and machine learning applications benefit tremendously from hardware acceleration, but existing compilers target either FPGAs, which sacrifice power and performanc…