Scalable task-based algorithm for multiplication of block-rank-sparse matrices

By Justus A. Calvin, C. Lewis, Edward F. Valeev. 2015

46 citations recorded by the source corpus.

A task-based formulation of Scalable Universal Matrix Multiplication Algorithm (SUMMA), a popular algorithm for matrix multiplication (MM), is applied to the multiplication of hierarchy-free, rank-structured matrices that appear in the domain of quantum chemistry (QC). The novel features of our formulation are: (1) concurrent scheduling of multiple SUMMA iterations, and (2) fine-grained task-based composition. These features make it tolerant of the load imbalance due to the irregular matrix structure and eliminate all artifactual sources of global synchronization. Scalability of iterative computation of square-root inverse of block-rank-sparse QC matrices is demonstrated; for full-rank (dense) matrices the performance of our SUMMA formulation usually exceeds that of the state-of-the-art dense MM implementations (ScaLAPACK and Cyclops Tensor Framework).

Subfields: Computational Theory and Mathematics, Computer Networks and Communications, ML Systems & Hardware, Numerical Methods & Scientific Computing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computational Theory and Mathematics

  1. Cliff Young
  2. Dustin Tran

In Computer Networks and Communications

  1. Penporn Koanantakool

In ML Systems & Hardware

  1. Noam Shazeer
  2. Blake A. Hechtman
  3. HyoukJoong Lee
  4. Ashish Vaswani

Citations in the corpus