LoopStack: a Lightweight Tensor Algebra Compiler Stack
Published in arXiv (Cornell University) • May 2, 2022
NobleIDNI6P19W68R35S36
Authors:,,
Bram Wasti
José Cambronero
Benoit Steiner
Abstract
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 enables us to compile entire neural networks and generate code targeting the AVX2, AVX512, NEON, and NEONfp16 instruction s...
Finding related papers...
Discussions
(0)No comments yet
Be the first to share your thoughts!