16.2k citations · 16.5k across the 9 of their papers we have counts for
Showing 2022Show all
2 papers · 1 filter
cs.LG2022★ 1 cited
OLLA: Optimizing the Lifetime and Location of Arrays to Reduce the Memory Usage of Neural Networks
Benoit Steiner, Mostafa Elhoushi, Jacob Kahn +1
The size of deep neural networks has grown exponentially in recent years. Unfortunately, hardware devices have not kept pace with the rapidly increasing memory requirements. To cop…
cs.LG2022★ 2 cited
LoopStack: a Lightweight Tensor Algebra Compiler Stack
Bram Wasti, José Pablo Cambronero, Benoit Steiner +2
We present LoopStack, a domain specific compiler stack for tensor operations, composed of a frontend, LoopTool, and an efficient optimizing code generator, LoopNest. This stack ena…