1 Hit in 1.9 sec

LoopStack: a Lightweight Tensor Algebra Compiler Stack [article]

Bram Wasti, José Pablo Cambronero, Benoit Steiner, Hugh Leather, Aleksandar Zlateski
2022 arXiv   pre-print
We present LoopStack, a domain specific compiler stack for tensor operations, composed of a frontend, LoopTool, and an efficient optimizing code generator, LoopNest.  ...  We evaluate our stack on a collection of full neural networks and commonly used network blocks as well as individual operators, and show that LoopStack generates machine code that matches and frequently  ...  Conclusion We presented LoopStack, a lightweight compiler designed from the ground up for deep learning workloads.  ... 
arXiv:2205.00618v1 fatcat:tk7gbuxl6feczkx36khx5s3bmu