Compiler Support for Sparse Tensor Computations in MLIR

التفاصيل البيبلوغرافية
العنوان: Compiler Support for Sparse Tensor Computations in MLIR
المؤلفون: Bik, Aart J. C., Koanantakool, Penporn, Shpeisman, Tatiana, Vasilache, Nicolas, Zheng, Bixia, Kjolstad, Fredrik
سنة النشر: 2022
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Programming Languages
الوصف: Sparse tensors arise in problems in science, engineering, machine learning, and data analytics. Programs that operate on such tensors can exploit sparsity to reduce storage requirements and computational time. Developing and maintaining sparse software by hand, however, is a complex and error-prone task. Therefore, we propose treating sparsity as a property of tensors, not a tedious implementation task, and letting a sparse compiler generate sparse code automatically from a sparsity-agnostic definition of the computation. This paper discusses integrating this idea into MLIR.
نوع الوثيقة: Working Paper
DOI: 10.1145/3544559
URL الوصول: http://arxiv.org/abs/2202.04305
رقم الأكسشن: edsarx.2202.04305
قاعدة البيانات: arXiv