In this paper, we introduce the Dynamic Load-balanced Tensor Contractions (DLTC), a domain-specific library for efficient task parallel execution of tensor contraction expressions, a class of computation encountered in quantum chemistry and physics. Our framework decomposes each contraction into smaller unit of tasks, represented by an abstraction referred to as iterators. We exploit an extra level of parallelism by having tasks across independent contractions executed concurrently through a dynamic load balancing run- time. We demonstrate the improved performance, scalability, and flexibility for the computation of tensor contraction expressions on parallel computers using examples from coupled cluster methods.
Revised: February 4, 2016 |
Published: November 17, 2013
Citation
Lai P., K. Stock, S. Rajbhandari, S. Krishnamoorthy, and P. Sadayappan. 2013.A Framework for Load Balancing of Tensor Contraction Expressions via Dynamic Task Partitioning. In Proceedings of the International Conference for High Performance Computing, Networking, Storage and Analysis (SC'13), November 17-21, 2013, Paper No. 13. New York, New York:Association for Computing Machinery (ACM).PNNL-SA-97996.doi:10.1145/2503210.2503290