WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program-Reference-Cited by-同舟云学术

WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program

Published:2023-01-27 Issue: Volume: Page:
ISSN:
Container-title:Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2
language:
Short-container-title:

Author:

Won Jaeyeon¹,Mendis Charith²,Emer Joel S.³,Amarasinghe Saman¹

Affiliation:

1. Massachusetts Institute of Technology, USA

2. University of Illinois at Urbana-Champaign, USA

3. Massachusetts Institute of Technology, USA / NVIDIA, USA

Funder

U.S.Department of Energy, Office of Science, Office of Advanced Scientific Computing Research

DARPA

NSF (National Science Foundation)

Publisher

ACM

Link

https://dl.acm.org/doi/pdf/10.1145/3575693.3575742

Reference52 articles.

1. Learning to optimize halide with tree search and random programs

2. Martin Alnæs , Jan Blechta , Johan Hake , August Johansson , Benjamin Kehlet , Anders Logg , Chris Richardson , Johannes Ring , Marie E Rognes , and Garth N Wells . 2015. The FEniCS Project Version 1.5. Archive of Numerical Software , Vol 3 ( 2015 ). Martin Alnæs, Jan Blechta, Johan Hake, August Johansson, Benjamin Kehlet, Anders Logg, Chris Richardson, Johannes Ring, Marie E Rognes, and Garth N Wells. 2015. The FEniCS Project Version 1.5. Archive of Numerical Software, Vol 3 (2015).

3. OpenTuner

4. Riyadh Baghdadi , Massinissa Merouani , Mohamed-Hicham Leghettas , Kamel Abdous , Taha Arbaoui , and Karima Benatchba . 2021 . A Deep Learning Based Cost Model for Automatic Code Optimization . Proceedings of Machine Learning and Systems , 3 (2021). Riyadh Baghdadi, Massinissa Merouani, Mohamed-Hicham Leghettas, Kamel Abdous, Taha Arbaoui, and Karima Benatchba. 2021. A Deep Learning Based Cost Model for Automatic Code Optimization. Proceedings of Machine Learning and Systems, 3 (2021).

Cited by 8 articles. 订阅此论文施引文献订阅此论文施引文献，注册后可以免费订阅5篇论文的施引文献，订阅后可以查看论文全部施引文献

1. Mind the Gap: Attainable Data Movement and Operational Intensity Bounds for Tensor Algorithms;2024 ACM/IEEE 51st Annual International Symposium on Computer Architecture (ISCA);2024-06-29

2. Minimum Cost Loop Nests for Contraction of a Sparse Tensor with a Tensor Network;Proceedings of the 36th ACM Symposium on Parallelism in Algorithms and Architectures;2024-06-17

3. Two-Face: Combining Collective and One-Sided Communication for Efficient Distributed SpMM;Proceedings of the 29th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2;2024-04-27

4. STile: Searching Hybrid Sparse Formats for Sparse Deep Learning Operators Automatically;Proceedings of the ACM on Management of Data;2024-03-12

5. Tetris: Accelerating Sparse Convolution by Exploiting Memory Reuse on GPU;Proceedings of the 29th ACM SIGPLAN Annual Symposium on Principles and Practice of Parallel Programming;2024-02-20