The Past, Present and Future of High Performance Numerical Linear Algebra: Minimizing Communication Jim Demmel EECS & Math Departments UC Berkeley.

Slides:



Advertisements
Similar presentations
Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
Advertisements

Statistical Modeling of Feedback Data in an Automatic Tuning System Richard Vuduc, James Demmel (U.C. Berkeley, EECS) Jeff.
Dense Linear Algebra (Data Distributions) Sathish Vadhiyar.
Load Balancing Parallel Applications on Heterogeneous Platforms.
Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
Parallel Jacobi Algorithm Steven Dong Applied Mathematics.
Eigenvalue and eigenvectors  A x = λ x  Quantum mechanics (Schrödinger equation)  Quantum chemistry  Principal component analysis (in data mining)
Communication Avoiding Algorithms for Dense Linear Algebra: LU, QR, and Cholesky decompositions and Sparse Cholesky decompositions Jim Demmel and Oded.
MATH 685/ CSI 700/ OR 682 Lecture Notes
Solving Linear Systems (Numerical Recipes, Chap 2)
Copyright © The McGraw-Hill Companies, Inc. Permission required for reproduction or display. Parallel Programming in C with MPI and OpenMP Michael J. Quinn.
CS 140 : Matrix multiplication Linear algebra problems Matrix multiplication I : cache issues Matrix multiplication II: parallel issues Thanks to Jim Demmel.
Avoiding Communication in Numerical Linear Algebra Jim Demmel EECS & Math Departments UC Berkeley.
Modern iterative methods For basic iterative methods, converge linearly Modern iterative methods, converge faster –Krylov subspace method Steepest descent.
Avoiding Communication in Linear Algebra Jim Demmel UC Berkeley bebop.cs.berkeley.edu.
Symmetric Eigensolvers in Sca/LAPACK Osni Marques
Avoiding Communication in Sparse Iterative Solvers Erin Carson Nick Knight CS294, Fall 2011.
Minisymposia 9 and 34: Avoiding Communication in Linear Algebra Jim Demmel UC Berkeley bebop.cs.berkeley.edu.
Avoiding Communication in Linear Algebra Jim Demmel UC Berkeley bebop.cs.berkeley.edu.
The Landscape of Ax=b Solvers Direct A = LU Iterative y’ = Ay Non- symmetric Symmetric positive definite More RobustLess Storage (if sparse) More Robust.
High Performance Computing 1 Parallelization Strategies and Load Balancing Some material borrowed from lectures of J. Demmel, UC Berkeley.
CS 240A: Solving Ax = b in parallel °Dense A: Gaussian elimination with partial pivoting Same flavor as matrix * matrix, but more complicated °Sparse A:
Introduction to Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
CS240A: Conjugate Gradients and the Model Problem.
Linear Algebra on GPUs Vasily Volkov. GPU Architecture Features SIMD architecture – Don’t be confused by scalar ISA which is only a program model We use.
CS294: Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments Oded Schwartz EECS Department.
MATH 685/ CSI 700/ OR 682 Lecture Notes Lecture 6. Eigenvalue problems.
Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley.
Introduction to Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
Communication avoiding algorithms in dense linear algebra
Communication-Avoiding Algorithms for Linear Algebra and Beyond
Implementing Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
COALA: Communication Optimal Algorithms for Linear Algebra Jim Demmel EECS & Math Depts. UC Berkeley Laura Grigori INRIA Saclay Ile de France.
1 Minimizing Communication in Numerical Linear Algebra Introduction, Technological Trends Jim Demmel EECS & Math Departments,
How to Compute and Prove Lower and Upper Bounds on the Communication Costs of Your Algorithm Part III: Graph analysis Oded Schwartz CS294, Lecture #10.
1 Jim Demmel EECS & Math Departments, UC Berkeley Minimizing Communication in Numerical Linear Algebra
1 Minimizing Communication in Linear Algebra James Demmel 15 June
The Future of LAPACK and ScaLAPACK Jim Demmel UC Berkeley CScADS Autotuning Workshop 9 July 2007.
Sparse Matrix-Vector Multiplication on Throughput-Oriented Processors
1 Minimizing Communication in Numerical Linear Algebra Optimizing One-sided Factorizations: LU and QR Jim Demmel EECS & Math.
03/04/2009CS267 Lecture 12a1 CS 267 Dense Linear Algebra: Possible Class Projects James Demmel
Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley.
Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel Math & EECS Departments UC Berkeley.
CS240A: Conjugate Gradients and the Model Problem.
Case Study in Computational Science & Engineering - Lecture 5 1 Iterative Solution of Linear Systems Jacobi Method while not converged do { }
Minimizing Communication in Numerical Linear Algebra Optimizing Krylov Subspace Methods Jim Demmel EECS & Math Departments,
Linear Algebra Libraries: BLAS, LAPACK, ScaLAPACK, PLASMA, MAGMA
Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley.
High Performance LU Factorization for Non-dedicated Clusters Toshio Endo, Kenji Kaneda, Kenjiro Taura, Akinori Yonezawa (University of Tokyo) and the future.
Introduction to Communication-Avoiding Algorithms Jim Demmel EECS & Math Departments UC Berkeley.
Report from LBNL TOPS Meeting TOPS/ – 2Investigators  Staff Members:  Parry Husbands  Sherry Li  Osni Marques  Esmond G. Ng 
Krylov-Subspace Methods - I Lecture 6 Alessandra Nardi Thanks to Prof. Jacob White, Deepak Ramaswamy, Michal Rewienski, and Karen Veroy.
How to Compute and Prove Lower and Upper Bounds on the Communication Costs of Your Algorithm Part II: Geometric embedding Oded Schwartz CS294, Lecture.
Potential Projects Jim Demmel CS294 Fall, 2011 Communication-Avoiding Algorithms
Performance of BLAS-3 Based Tridiagonalization Algorithms on Modern SMP Machines Yusaku Yamamoto Dept. of Computational Science & Engineering Nagoya University.
09/13/2012CS4230 CS4230 Parallel Programming Lecture 8: Dense Linear Algebra and Locality Optimizations Mary Hall September 13,
Linear Algebra Libraries: BLAS, LAPACK, ScaLAPACK, PLASMA, MAGMA Shirley Moore CPS5401 Fall 2013 svmoore.pbworks.com November 12, 2012.
Algorithms for Supercomputers Upper bounds: from sequential to parallel Oded Schwartz Seminar: Sunday, 12-2pm, B410 Workshop: Sunday, 2-5pm High performance.
Conjugate gradient iteration One matrix-vector multiplication per iteration Two vector dot products per iteration Four n-vectors of working storage x 0.
Algorithms for Supercomputers Introduction Oded Schwartz Seminar: Sunday, 12-2pm, B410 Workshop: ? High performance Fault tolerance
LLNL-PRES This work was performed under the auspices of the U.S. Department of Energy by Lawrence Livermore National Laboratory under Contract DE-AC52-07NA27344.
Communication-Avoiding Iterative Methods
Model Problem: Solving Poisson’s equation for temperature
Minimizing Communication in Linear Algebra
Minimizing Communication in Linear Algebra
BLAS: behind the scenes
P A R A L L E L C O M P U T I N G L A B O R A T O R Y
Avoiding Communication in Linear Algebra
Dense Linear Algebra (Data Distributions)
Presentation transcript:

The Past, Present and Future of High Performance Numerical Linear Algebra: Minimizing Communication Jim Demmel EECS & Math Departments UC Berkeley

The Past, the Present … LAPACK – sequential/shared memory parallel dense linear algebra library

The Past, the Present … LAPACK contributors – Jack Dongarra, Julien Langou, Julie Langou, Ed Anderson, Zhaojun Bai, David Bailey, David Bindel, Chris Bischof, Susan Blackford, Zvonomir Bujanovic, Karen Braman, Ralph Byers, Inderjit Dhillon, Zlatko Drmac, Peng Du, Jeremy Du Croz, Mark Fahey, Anne Greenbaum, Ming Gu, Fred Gustavson, Deaglan Halligan, Sven Hammarling, Greg Henry, Yozo Hida, Nick Higham, Bo Kagstrom, William Kahan, Daniel Kressner, Ren-Cang Li, Xiaoye Li, Craig Lucas, Osni Marques, Peter Mayes, Alan McKenney, Beresford Parlett, Antoine Petitet, Peter Poromaa, Enrique Quintana-Orti, Gregorio Quintana-Orti, Giuseppe Radicati, Huan Ren, Jason Riedy, Jeff Rutter, Danny Sorensen, Ken Stanley, Xiaobai Sun, Brian Sutton, Francoise Tisseur, Robert van de Geijn, Kresimir Veselic, Christof Voemel, Jerzy Wasniewski + many undergrads

The Past, the Present … LAPACK – seq/shared mem dense linear algebra ScaLAPACK – distributed memory parallel dense linear algebra library – Jack Dongarra, Susan Blackford, Jaeyoung Choi, Andy Cleary, Ed D’Azevedo, Inderjit Dhillon, Sven Hammarling, Greg Henry, Antoine Petitet, Ken Stanley, David Walker, Clint Whaley, and many other contributors

The Past, the Present … LAPACK – seq/shared mem dense linear algebra ScaLAPACK – dist mem dense linear algebra SuperLU – sparse direct solver for Ax=b – Xiaoye Sherry Li

The Past, the Present … LAPACK – seq/shared mem dense linear algebra ScaLAPACK – dist mem dense linear algebra SuperLU – sparse direct solver for Ax=b Autotuning – PhiPAC for matrix multiplication Jeff Bilmes, Krste Asanovic, Rich Vuduc, Sriram Iyer, CheeWhye Chin, Dominic Lam – OSKI for sparse-matrix-vector multiplication (SpMV) Rich Vuduc, Kathy Yelick, Rajesh Nishtala, Ben Lee, Shoaib Kamil, Jen Hsu

The Past, the Present … LAPACK – seq/shared mem dense linear algebra ScaLAPACK – dist mem dense linear algebra SuperLU – sparse direct solver for Ax=b Autotuning – matmul and SpMV Prometheus – parallel unstructured FE solver – Mark Adams – Gordon Bell Prize, 2004

… The Future Why we need to “avoid communication,” i.e. avoid moving data “Direct” Linear Algebra – Lower bounds on communication for linear algebra problems like Ax=b, least squares, Ax = λx, SVD, etc – New algorithms that attain these lower bounds Not in libraries like Sca/LAPACK (yet!) Large speed-ups possible “Iterative” Linear Algebra – Ditto for Krylov Subspace Methods

9 Why avoid communication? (1/2) Algorithms have two costs: 1.Arithmetic (FLOPS) 2.Communication: moving data between – levels of a memory hierarchy (sequential case) – processors over a network (parallel case). CPU Cache DRAM CPU DRAM CPU DRAM CPU DRAM CPU DRAM

Why avoid communication? (2/2) Running time of an algorithm is sum of 3 terms: – # flops * time_per_flop – # words moved / bandwidth – # messages * latency 10 communication Time_per_flop << 1/ bandwidth << latency Gaps growing exponentially with time (FOSC, 2004) Goal : reorganize linear algebra to avoid communication Between all memory hierarchy levels L1 L2 DRAM network, etc Not just hiding communication (speedup  2x ) Arbitrary speedups possible Annual improvements Time_per_flopBandwidthLatency Network26%15% DRAM23%5% 59%

Direct linear algebra: Prior Work on Matmul Assume n 3 algorithm for C=A*B (i.e. not Strassen-like) Sequential case, with fast memory of size M – Lower bound on #words moved to/from slow memory =  (n 3 / M 1/2 ) [Hong, Kung, 81] – Attained using “blocked” or cache-oblivious algorithms 11 Parallel case on P processors: Assume load balanced, one copy each of A, B, C Lower bound on #words communicated =  ( n 2 / P 1/2 ) [Irony, Tiskin, Toledo, 04] Attained by Cannon’s Algorithm NNZLower bound on #words Attained by 3n 2 (“2D alg”)  (n 2 / P 1/2 ) [Cannon, 69] 3n 2 P 1/3 (“3D alg”)  (n 2 / P 2/3 ) [Johnson,93]

Lower bound for all “direct” linear algebra Holds for – BLAS, LU, QR, eig, SVD, tensor contractions, … – Some whole programs (sequences of these operations, no matter how individual ops are interleaved, eg A k ) – Dense and sparse matrices (where #flops << n 3 ) – Sequential and parallel algorithms – Some graph-theoretic algorithms (eg Floyd-Warshall) 12 Let M = “fast” memory size (per processor) #words_moved (per processor) =  (#flops (per processor) / M 1/2 ) #messages_sent (per processor) =  (#flops (per processor) / M 3/2 ) Parallel case: assume either load or memory balanced

Can we attain these lower bounds? Do conventional dense algorithms as implemented in LAPACK and ScaLAPACK attain these bounds? – Mostly not If not, are there other algorithms that do? – Yes, for dense linear algebra Only a few sparse algorithms so far (eg Cholesky) 13

One-sided Factorizations (LU, QR), so far Classical Approach for i=1 to n update column i update trailing matrix #words_moved = O(n 3 ) 14 Blocked Approach (LAPACK) for i=1 to n/b update block i of b columns update trailing matrix #words moved = O(n 3 /M 1/3 ) Recursive Approach func factor(A) if A has 1 column, update it else factor(left half of A) update right half of A factor(right half of A) #words moved = O(n 3 /M 1/2 ) None of these approaches minimizes #messages Need another idea

TSQR: QR of a Tall, Skinny matrix 15 W = Q 00 R 00 Q 10 R 10 Q 20 R 20 Q 30 R 30 W0W1W2W3W0W1W2W3 Q 00 Q 10 Q 20 Q 30 = =. R 00 R 10 R 20 R 30 R 00 R 10 R 20 R 30 = Q 01 R 01 Q 11 R 11 Q 01 Q 11 =. R 01 R 11 R 01 R 11 = Q 02 R 02

TSQR: QR of a Tall, Skinny matrix 16 W = Q 00 R 00 Q 10 R 10 Q 20 R 20 Q 30 R 30 W0W1W2W3W0W1W2W3 Q 00 Q 10 Q 20 Q 30 = =. R 00 R 10 R 20 R 30 R 00 R 10 R 20 R 30 = Q 01 R 01 Q 11 R 11 Q 01 Q 11 =. R 01 R 11 R 01 R 11 = Q 02 R 02

Minimizing Communication in TSQR W = W0W1W2W3W0W1W2W3 R 00 R 10 R 20 R 30 R 01 R 11 R 02 Parallel: W = W0W1W2W3W0W1W2W3 R 01 R 02 R 00 R 03 Sequential: W = W0W1W2W3W0W1W2W3 R 00 R 01 R 11 R 02 R 11 R 03 Dual Core: Can choose reduction tree dynamically Multicore / Multisocket / Multirack / Multisite / Out-of-core: ?

TSQR Performance Results Parallel – Intel Clovertown – Up to 8x speedup (8 core, dual socket, 10M x 10) – Pentium III cluster, Dolphin Interconnect, MPICH Up to 6.7x speedup (16 procs, 100K x 200) – BlueGene/L Up to 4x speedup (32 procs, 1M x 50) – Grid – 4x on 4 cities (Dongarra et al) – Cloud – early result – up and running using Mesos Sequential – “Infinite speedup” for out-of-Core on PowerPC laptop As little as 2x slowdown vs (predicted) infinite DRAM LAPACK with virtual memory never finished What about QR for a general matrix? 18 Data from Grey Ballard, Mark Hoemmen, Laura Grigori, Julien Langou, Jack Dongarra, Michael Anderson

CAQR on a GPU (Fermi C2050) (1/2) # rows = 8192

CAQR on a GPU (Fermi C2050) (2/2) #rows = 8192 Biggest speedup over MAGMA QR is 13x for 1M x 192

Partial History of dense sequential QR algorithms attaining communication lower bounds 21 Algorithms shown minimizing # Messages assume (recursive) block layout Many references (see reports), only some shown, plus ours Oldest reference may or may not include analysis Cache-oblivious are underlined, Green are ours, ? is unknown/future work b = block size, M = fast memory size, n = dimension Algorithm2 Levels of MemoryMultiple Levels of Memory #Words Movedand # Messages#Words Movedand # Messages QR[Elmroth, Gustavson,98] Recursive, may do more flops LAPACK (only for n  M and b=M 1/2, or n 2  M) [DGHL08] [Frens,Wise,03] explicit Q only, 3x more flops [DGHL08] implicit Q, but represented differently, ~ same flop count [Elmroth, Gustavson,98] [DGHL08]? [Frens,Wise,03] [DGHL08]? Can we get the same flop count?

CALU: Using similar idea for TSLU as TSQR: Use reduction tree, to do “Tournament Pivoting” 22 W nxb = W1W2W3W4W1W2W3W4 P 1 ·L 1 ·U 1 P 2 ·L 2 ·U 2 P 3 ·L 3 ·U 3 P 4 ·L 4 ·U 4 = Choose b pivot rows of W 1, call them W 1 ’ Choose b pivot rows of W 2, call them W 2 ’ Choose b pivot rows of W 3, call them W 3 ’ Choose b pivot rows of W 4, call them W 4 ’ W1’W2’W3’W4’W1’W2’W3’W4’ P 12 ·L 12 ·U 12 P 34 ·L 34 ·U 34 = Choose b pivot rows, call them W 12 ’ Choose b pivot rows, call them W 34 ’ W 12 ’ W 34 ’ = P 1234 ·L 1234 ·U 1234 Choose b pivot rows Go back to W and use these b pivot rows (move them to top, do LU without pivoting) Repeat on each set of b columns to do CALU

Stability of CALU using Tournament Pivoting Thm: Tournament Pivoting (TP) as stable as Partial Pivoting (PP) in following sense: TP gets same results as PP applied to different input matrix whose entries are blocks taken from input A Worst case pivot growth factor on n x n matrix – Proven: 2 nH-1 where H = 1 + height of reduction tree – Attained: 2 n-1, same as PP There are examples where PP is exponentially less stable than TP, and vice-versa, so neither is always better than the other Extensive numerical testing confirms stability Summer School Lecture 4 23

Exascale Machine Parameters 2^30  1,000,000 nodes 1024 cores/node (a billion cores!) 100 GB/sec interconnect bandwidth 400 GB/sec DRAM bandwidth 1 microsec interconnect latency 50 nanosec memory latency 32 Petabytes of memory 1/2 GB total L1 on a node 20 Megawatts !?

Exascale predicted speedups for CA-LU vs ScaLAPACK-LU log 2 (p) log 2 (n 2 /p) = log 2 (memory_per_proc)

History of Spectral Divide-and-Conquer for Eigenvalue problems and the SVD Ideas go back to Bulgakov, Godunov, Malyshev [BG88,Mal89] Bai, D., Gu [BDG97] – Reduced to matmul, QR, generalized QR with pivoting (bug) D., Dumitriu, Holtz [DDH07] – Uses RURV (randomized URV), not pivoted QR (same bug) D., Grigori, Hoemmen, Langou [DGHL08] – Communication-avoiding QR (CAQR) New Communication-avoiding algorithm [BDD10] – Use generalized RURV for better rank-detection (fixes bug) – Uses communication-avoiding matmul and QR – Uses randomization in divide-and-conquer – Ballard, D., Dumitriu 26

Overview of Spectral Divide-and-Conquer Implicitly compute (I + A -2 k ) -1 – Maps spectrum to 0 and 1 (as k   ) – Converges to spectral projector of A for |λ| > 1 – Rank Revealing QR yields invariant subspace 27 A new = Q*AQ = A 11 A 12  A 22 A 0 = A, B 0 = I Repeat A k+1 = Q 12 *  A k B k+1 = Q 22 *  B k until R k converges Q 11 Q 12  R k = qr B k Q 12 Q 22 0 A k Fact: A k -1 B k = A -2 k So need RRQR of (I + A -2 k ) -1 = (A k + B k ) -1 A k without forming (A k + B k ) -1 A k Algorithm:

Rank Revealing CA-Decompositions QR with Tournament Pivoting – Emulate column pivoting by using a ``tournament” to pick next group of b columns – Only applies to single matrix, not (A k + B k ) -1 A k 28 Randomized URV of A [V,R’] = caqr(randn(n,n)) … V random orthogonal [U,R] = caqr(A·V T ) Thm: A = URV reveals the rank of A with high probability Randomized URV of (A k + B k ) -1 A k [V,R’] = caqr(randn(n,n)) … V random orthogonal [Q 1,R 1 ] = caqr(A k ·V T ) [R 2,U] = carq(Q 1 T ·(A k + B k )) Thm: (A k + B k ) -1 A k = U T (R 2 -1 R 1 )V reveals rank with high probability Only need U, not R 2 -1 R 1 Applies to M 1  1 · M 2  1 · M 3  1 · · · = URV

Splitting the spectrum in different places Splitting the spectrum along unit circle not enough Nonsymmetric case – Moebius transformation z  (  z +  )/(  z +  ) maps unit circle to any line or circle in complex plane – Apply splitting algorithm to (A 0,B 0 ) = (  A +  I,  A +  I) instead of (A,I), to split along any desired line or circle Symmetric case - easier – Just shift A to A -  I – SVD analogous 29

Randomized Spectral Divide-and-Conquer Pick random place to divide spectrum Two progress metrics – Spectrum splits, with f  n and (1-f)  n evals per part, with 0 < f < 1 – Region splits, with fractions f and 1-f of area in each – Progress guaranteed with high probability; always stable SymEig and SVD: random point near center of spectrum NonsymEig: random line/circle near “center” of spectrum 30

Successive Band Reduction Works just for eig(A=A T ) or SVD – Bischof, Lang Sun Step 1: Reduce Dense  Narrow Band – If bandwidth  M 1/2, can minimize communication Step 2: Reduce Narrow Band  Tridiagonal – Trickier… 31

Q5TQ5T Q1Q1 Q1TQ1T Q2Q2 Q2TQ2T Q3Q3 Q3TQ3T Q5Q5 Q4Q4 Q4TQ4T b+1 d+1 c c d+c Successive Band Reduction To minimize communication (when n > M 1/2 ) after reducing to bandwidth b = M 1/2 : if n > M Use d=b-1 and c=1 else if M 1/2 < n ≤ M Pass 1: Use d=b – M/(4n) and c=M/(4n) Pass 2: Use d=M/(4n)-1 and c=1

Summary of dense parallel algorithms attaining communication lower bounds 33 Assume nxn matrices on P processors, memory per processor = O(n 2 / P) ScaLAPACK assumes best block size b chosen Many references (see reports), Green are ours Recall lower bounds: #words_moved =  ( n 2 / P 1/2 ) and #messages =  ( P 1/2 ) AlgorithmReferenceFactor exceeding lower bound for #words_moved Factor exceeding lower bound for #messages Matrix multiply[Cannon, 69]11 CholeskyScaLAPACKlog P LU[GDX08] ScaLAPACK log P ( N / P 1/2 ) · log P QR[DGHL08] ScaLAPACK log P log 3 P ( N / P 1/2 ) · log P Sym Eig, SVD[BDD10] ScaLAPACK log P log 3 P N / P 1/2 Nonsym Eig[BDD10] ScaLAPACK log P P 1/2 · log P log 3 P N · log P Can we do better?

Summary of dense parallel algorithms attaining communication lower bounds 34 Assume nxn matrices on P processors, memory per processor = O(n 2 / P)? Why? ScaLAPACK assumes best block size b chosen Many references (see reports), Green are ours Recall lower bounds: #words_moved =  ( n 2 / P 1/2 ) and #messages =  ( P 1/2 ) AlgorithmReferenceFactor exceeding lower bound for #words_moved Factor exceeding lower bound for #messages Matrix multiply[Cannon, 69]11 CholeskyScaLAPACKlog P LU[GDX08] ScaLAPACK log P ( N / P 1/2 ) · log P QR[DGHL08] ScaLAPACK log P log 3 P ( N / P 1/2 ) · log P Sym Eig, SVD[BDD10] ScaLAPACK log P log 3 P N / P 1/2 Nonsym Eig[BDD10] ScaLAPACK log P P 1/2 · log P log 3 P N · log P Can we do better?

Beating #words_moved =  (n 2 /P 1/2 ) 35 i j k “A face” “B face” “C face” A(2,1) A(1,3) B(1,3) B(3,1) C(1,1) C(2,3) Cube representing C(1,1) += A(1,3)*B(3,1) Johnson’s Matmul Algorithm on P 1/3 x P 1/3 x P 1/3 processor grid Broadcast A in j direction (P 1/3 redundant copies) Broadcast B in i direction (ditto) Local multiplies Reduce (sum) in k direction Communication volume = O( (n/P 1/3 ) 2 log(P) ) - optimal Number of messages = O(log(P)) – optimal Extends to 1 ≤ c ≤ P 1/3 copies What about LU, etc? #words_moved =  ((n 3 /P)/local_mem 1/2 ) If one copy of data, local_mem = n 2 /P Can we use more memory to communicate less?

Predicted Exascale 2.5D Matmul - Speedup over Cannon log 2 (n 2 /p) = log 2 (memory_per_proc) log 2 (p)

Exascale 2.5D Matmul – optimal # copies log 2 (n 2 /p) = log 2 (memory_per_proc) log 2 (p)

38 Strassen-like Algorithms M = fast memory size Conventional matmul #flops = 2n 3 #words_moved =  (n 3 /M 1/2 ) =  (M(n/M 1/2 ) 3 ), attainable Strassen matmul #flops =  (n  ) where  = log 2 (7)  2.81 #words_moved =  (M(n/M 1/2 )  ), attainable too Applies to other sufficiently “Strassen-like“ algorithms How broadly does it apply? We know rest of linear algebra can be done in O(n  ) flops Also with #words_moved =  (M(n/M 1/2 )  ) (sequential) Lower bounds?

Summary of Direct Linear Algebra New lower bounds, optimal algorithms, big speedups in theory and practice Lots of other progress, open problems – New ways to “pivot” – Heterogeneous architectures – Some sparse algorithms – Autotuning...

Avoiding Communication in Iterative Linear Algebra k-steps of iterative solver for sparse Ax=b or Ax=λx – Does k SpMVs with A and starting vector – Many such “Krylov Subspace Methods” Conjugate Gradients (CG), GMRES, Lanczos, Arnoldi, … Goal: minimize communication – Assume matrix “well-partitioned” – Serial implementation Conventional: O(k) moves of data from slow to fast memory New: O(1) moves of data – optimal – Parallel implementation on p processors Conventional: O(k log p) messages (k SpMV calls, dot prods) New: O(log p) messages - optimal Lots of speed up possible (modeled and measured) – Price: some redundant computation 40

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3 Works for any “well-partitioned” A

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Example: A tridiagonal, n=32, k=3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Sequential Algorithm Example: A tridiagonal, n=32, k=3 Step 1

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Sequential Algorithm Example: A tridiagonal, n=32, k=3 Step 1Step 2

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Sequential Algorithm Example: A tridiagonal, n=32, k=3 Step 1Step 2Step 3

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Sequential Algorithm Example: A tridiagonal, n=32, k=3 Step 1Step 2Step 3Step 4

… … 32 x A·x A 2 ·x A 3 ·x Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x] Replace k iterations of y = A  x with [Ax, A 2 x, …, A k x] Parallel Algorithm Example: A tridiagonal, n=32, k=3 Each processor works on (overlapping) trapezoid Proc 1Proc 2Proc 3Proc 4

Remotely Dependent Entries for [x,Ax,A 2 x,A 3 x], A irregular, multiple processors Communication Avoiding Kernels: The Matrix Powers Kernel : [Ax, A 2 x, …, A k x]

Minimizing Communication of GMRES to solve Ax=b GMRES: find x in span{b,Ab,…,A k b} minimizing || Ax-b || 2 Standard GMRES for i=1 to k w = A · v(i-1) … SpMV MGS(w, v(0),…,v(i-1)) update v(i), H endfor solve LSQ problem with H Communication-avoiding GMRES W = [ v, Av, A 2 v, …, A k v ] [Q,R] = TSQR(W) … “Tall Skinny QR” build H from R solve LSQ problem with H Oops – W from power method, precision lost! 53 Sequential case: #words moved decreases by a factor of k Parallel case: #messages decreases by a factor of k

“Monomial” basis [Ax,…,A k x] fails to converge Different polynomial basis [p 1 (A)x,…,p k (A)x] does converge 54

Speed ups of GMRES on 8-core Intel Clovertown [MHDY09] 55 Requires Co-tuning Kernels

Exascale predicted speedups for Matrix Powers Kernel over SpMV for 2D Poisson (5 point stencil) log 2 (p) log 2 (n 2 /p) = log 2 (memory_per_proc)

Summary of Iterative Linear Algebra New Lower bounds, optimal algorithms, big speedups in theory and practice Lots of other progress, open problems – GMRES, CG, BiCGStab, Arnoldi, Lanczos reorganized – Other Krylov methods? – Recognizing stable variants more easily? – Avoiding communication with preconditioning harder “Hierarchically semi-separable” preconditioners work – Autotuning

For further information Papers – bebop.cs.berkeley.edu – 1-week-short course – slides and video –

Collaborators and Supporters Collaborators – Michael Anderson (UCB), Grey Ballard (UCB), Erin Carson (UCB), Jack Dongarra (UTK), Ioana Dumitriu (U. Wash), Laura Grigori (INRIA), Ming Gu (UCB), Mark Hoemmen (Sandia NL), Olga Holtz (UCB & TU Berlin), Nick Knight (UCB), Julien Langou, (U Colo. Denver), Marghoob Mohiyuddin (UCB), Oded Schwartz (TU Berlin), Edgar Solomonik (UCB), Michelle Strout (Colo. SU), Vasily Volkov (UCB), Sam Williams (LBNL), Hua Xiang (INRIA), Kathy Yelick (UCB & LBNL) – Other members of the ParLab, BEBOP, CACHE, EASI, MAGMA, PLASMA, TOPS projects Supporters – NSF, DOE, UC Discovery – Intel, Microsoft, Mathworks, National Instruments, NEC, Nokia, NVIDIA, Samsung, Sun

We’re hiring! Seeking software engineers to help develop the next versions of LAPACK and ScaLAPACK Locations: UC Berkeley and UC Denver Please send applications to

Summary Don’t Communic… 61 Time to redesign all linear algebra algorithms and software