Ameer Haj Ali

TenSet: A Large-scale Program Performance Dataset for Learned Tensor Compilers

Search-based tensor compilers can greatly accelerate the execution of machine learning models by generating high-performance tensor programs, such as matrix multiplications and convolutions. These compilers take a high-level mathematical expression as input and search for the fastest low-level imple…