communication-avoiding algorithms for linear algebra and beyond jim demmel eecs & math...

67
Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Upload: tamsin-austin

Post on 24-Dec-2015

217 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Communication-Avoiding Algorithmsfor Linear Algebra and Beyond

Jim DemmelEECS & Math Departments

UC Berkeleybebop.cs.berkeley.edu

Page 2: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

2

Why avoid communication? (1/2)

Algorithms have two costs (measured in time or energy):1. Arithmetic (FLOPS)2. Communication: moving data between

– levels of a memory hierarchy (sequential case) – processors over a network (parallel case).

CPUCache

DRAM

CPUDRAM

CPUDRAM

CPUDRAM

CPUDRAM

Page 3: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Why avoid communication? (2/2)

• Running time of an algorithm is sum of 3 terms:– # flops * time_per_flop– # words moved / bandwidth– # messages * latency

3

communication

• Time_per_flop << 1/ bandwidth << latency• Gaps growing exponentially with time [FOSC]

• Avoid communication to save time• Same story for energy

Annual improvements

Time_per_flop Bandwidth Latency

Network 26% 15%

DRAM 23% 5%59%

Page 4: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Goals

4

• Redesign algorithms to avoid communication• Between all memory hierarchy levels

• L1 L2 DRAM network, etc • Attain lower bounds if possible

• Current algorithms often far from lower bounds• Large speedups and energy savings possible

Page 5: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Sample Speedups

• Up to 12x faster for 2.5D matmul on 64K core IBM BG/P

• Up to 3x faster for tensor contractions on 2K core Cray XE/6

• Up to 6.2x faster for APSP on 24K core Cray CE6

• Up to 2.1x faster for 2.5D LU on 64K core IBM BG/P

• Up to 11.8x faster for direct N-body on 32K core IBM BG/P

• Up to 13x faster for TSQR on Tesla C2050 Fermi NVIDIA GPU

• Up to 6.7x faster for symeig (band A) on 10 core Intel Westmere

• Up to 2x faster for 2.5D Strassen on 38K core Cray XT4

• Up to 4.2x faster for MiniGMG benchmark bottom solver, using

CA-BiCGStab (2.5x for overall solve)

– 2.5x / 1.5x for combustion simulation code5

Page 6: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

“New Algorithm Improves Performance and Accuracy on Extreme-Scale Computing Systems. On modern computer architectures, communication between processors takes longer than the performance of a floating point arithmetic operation by a given processor. ASCR researchers have developed a new method, derived from commonly used linear algebra methods, to minimize communications between processors and the memory hierarchy, by reformulating the communication patterns specified within the algorithm. This method has been implemented in the TRILINOS framework, a highly-regarded suite of software, which provides functionality for researchers around the world to solve large scale, complex multi-physics problems.”

FY 2010 Congressional Budget, Volume 4, FY2010 Accomplishments, Advanced Scientific Computing Research (ASCR), pages 65-67.

President Obama cites Communication-Avoiding Algorithms in the FY 2012 Department of Energy Budget Request to Congress:

CA-GMRES (Hoemmen, Mohiyuddin, Yelick, JD)“Tall-Skinny” QR (Grigori, Hoemmen, Langou, JD)

Page 7: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural implications

Page 8: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods• Architectural Implications

Page 9: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Summary of CA Linear Algebra• “Direct” Linear Algebra

• Lower bounds on communication for linear algebra problems like Ax=b, least squares, Ax = λx, SVD, etc

• Mostly not attained by algorithms in standard libraries• New algorithms that attain these lower bounds

• Being added to libraries: Sca/LAPACK, PLASMA, MAGMA, …

• Large speed-ups possible• Autotuning to find optimal implementation

• Ditto for “Iterative” Linear Algebra

Page 10: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Lower bound for all “n3-like” linear algebra

• Holds for– Matmul, BLAS, LU, QR, eig, SVD, tensor contractions, …– Some whole programs (sequences of these operations, no

matter how individual ops are interleaved, eg Ak)– Dense and sparse matrices (where #flops << n3 )– Sequential and parallel algorithms– Some graph-theoretic algorithms (eg Floyd-Warshall)

10

• Let M = “fast” memory size (per processor)

#words_moved (per processor) = (#flops (per processor) / M1/2 )

#messages_sent (per processor) = (#flops (per processor) / M3/2 )

• Parallel case: assume either load or memory balanced

Page 11: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Lower bound for all “n3-like” linear algebra

• Holds for– Matmul, BLAS, LU, QR, eig, SVD, tensor contractions, …– Some whole programs (sequences of these operations, no

matter how individual ops are interleaved, eg Ak)– Dense and sparse matrices (where #flops << n3 )– Sequential and parallel algorithms– Some graph-theoretic algorithms (eg Floyd-Warshall)

11

• Let M = “fast” memory size (per processor)

#words_moved (per processor) = (#flops (per processor) / M1/2 )

#messages_sent ≥ #words_moved / largest_message_size

• Parallel case: assume either load or memory balanced

Page 12: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Lower bound for all “n3-like” linear algebra

• Holds for– Matmul, BLAS, LU, QR, eig, SVD, tensor contractions, …– Some whole programs (sequences of these operations, no

matter how individual ops are interleaved, eg Ak)– Dense and sparse matrices (where #flops << n3 )– 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) / M1/2 )

#messages_sent (per processor) = (#flops (per processor) / M3/2 )

• Parallel case: assume either load or memory balanced

SIAM SIAG/Linear Algebra Prize, 2012Ballard, D., Holtz, Schwartz

Page 13: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Can we attain these lower bounds?

• Do conventional dense algorithms as implemented in LAPACK and ScaLAPACK attain these bounds?– Often not

• If not, are there other algorithms that do?– Yes, for much of dense linear algebra– New algorithms, with new numerical properties,

new ways to encode answers, new data structures

– Not just loop transformations (need those too!)• Only a few sparse algorithms so far• Lots of work in progress

13

Page 14: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural Implications

Page 15: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

TSQR: QR of a Tall, Skinny matrix

15

W =

Q00 R00

Q10 R10

Q20 R20

Q30 R30

W0

W1

W2

W3

Q00

Q10

Q20

Q30

= = .

R00

R10

R20

R30

R00

R10

R20

R30

=Q01 R01

Q11 R11

Q01

Q11

= . R01

R11

R01

R11

= Q02 R02

Page 16: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

TSQR: QR of a Tall, Skinny matrix

16

W =

Q00 R00

Q10 R10

Q20 R20

Q30 R30

W0

W1

W2

W3

Q00

Q10

Q20

Q30

= = .

R00

R10

R20

R30

R00

R10

R20

R30

=Q01 R01

Q11 R11

Q01

Q11

= . R01

R11

R01

R11

= Q02 R02

Output = { Q00, Q10, Q20, Q30, Q01, Q11, Q02, R02 }

Page 17: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

TSQR: An Architecture-Dependent Algorithm

W =

W0

W1

W2

W3

R00

R10

R20

R30

R01

R11

R02Parallel:

W =

W0

W1

W2

W3

R01R02

R00

R03

Sequential:

W =

W0

W1

W2

W3

R00

R01

R01

R11

R02

R11

R03

Dual Core:

Can choose reduction tree dynamically

Multicore / Multisocket / Multirack / Multisite / Out-of-core: ?

Page 18: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

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)

– Tesla C 2050 / Fermi• Up to 13x (110,592 x 100)

– Grid – 4x on 4 cities vs 1 city (Dongarra, Langou et al)– Cloud – (Gleich and Benson) ~2 map-reduces

• 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

• SVD costs about the same• Joint work with Grigori, Hoemmen, Langou, Anderson, Ballard, Keutzer, others

18

Data from Grey Ballard, Mark Hoemmen, Laura Grigori, Julien Langou, Jack Dongarra, Michael Anderson

Page 19: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Summary of dense parallel algorithms attaining communication lower bounds

• Assume nxn matrices on P processors • Minimum Memory per processor = M = O(n2 / P)• Recall lower bounds:

#words_moved = ( (n3/ P) / M1/2 ) = ( n2 / P1/2 ) #messages = ( (n3/ P) / M3/2 ) = ( P1/2 )

• Does ScaLAPACK attain these bounds?• For #words_moved: mostly, except nonsym. Eigenproblem• For #messages: asymptotically worse, except Cholesky

• New algorithms attain all bounds, up to polylog(P) factors• Cholesky, LU, QR, Sym. and Nonsym eigenproblems, SVD

Can we do Better?

Page 20: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Can we do better?

• Aren’t we already optimal?• Why assume M = O(n2/p), i.e. minimal?

– Lower bound still true if more memory– Can we attain it?

Page 21: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural Implications

Page 22: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

2.5D Matrix Multiplication

• Assume can fit cn2/P data per processor, c > 1• Processors form (P/c)1/2 x (P/c)1/2 x c grid

c

(P/c)1/2

(P/c)1/2

Example: P = 32, c = 2

Page 23: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

2.5D Matrix Multiplication

• Assume can fit cn2/P data per processor, c > 1• Processors form (P/c)1/2 x (P/c)1/2 x c grid

k

j

iInitially P(i,j,0) owns A(i,j) and B(i,j) each of size n(c/P)1/2 x n(c/P)1/2

(1) P(i,j,0) broadcasts A(i,j) and B(i,j) to P(i,j,k)

(2) Processors at level k perform 1/c-th of SUMMA, i.e. 1/c-th of Σm A(i,m)*B(m,j)

(3) Sum-reduce partial sums Σm A(i,m)*B(m,j) along k-axis so P(i,j,0) owns C(i,j)

Page 24: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

2.5D Matmul on BG/P, 16K nodes / 64K cores

Page 25: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

2.5D Matmul on BG/P, 16K nodes / 64K coresc = 16 copies

Distinguished Paper Award, EuroPar’11 (Solomonik, D.)SC’11 paper by Solomonik, Bhatele, D.

12x faster

2.7x faster

Page 26: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Perfect Strong Scaling – in Time and Energy • Every time you add a processor, you should use its memory M too• Start with minimal number of procs: PM = 3n2

• Increase P by a factor of c total memory increases by a factor of c• Notation for timing model:

– γT , βT , αT = secs per flop, per word_moved, per message of size m

• T(cP) = n3/(cP) [ γT+ βT/M1/2 + αT/(mM1/2) ]

= T(P)/c• Notation for energy model:

– γE , βE , αE = joules for same operations

– δE = joules per word of memory used per sec

– εE = joules per sec for leakage, etc.

• E(cP) = cP { n3/(cP) [ γE+ βE/M1/2 + αE/(mM1/2) ] + δEMT(cP) + εET(cP) }

= E(P)• Perfect scaling extends to N-body, Strassen, …

Page 27: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural Implications

Page 28: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Sparse Matrices• If matrix quickly becomes dense, use dense algorithm

– Ex: All-Pairs-Shortest-Path, using Floyd-Warshall– Kleene’s Algorithm allows 2.5D optimizations– Up to 6.2x speedup on 1024 node Cray XE6, for n=4096

• If parts of the matrix become dense, optimize those– Ex: Cholesky on matrix with good separators

– Lower bound = Ω( w3 / M1/2 ), w = size of largest separator– Attained by PSPACES with optimal dense Cholesky on separators

• If matrix stays very sparse, lower bound unattainable, new one?– Ex: A*B, both are Erdos-Renyi: Prob(A(i,j)≠0) = d/n, d << n1/2, iid

– Thm: A parallel algorithm that is sparsity-independent and load balanced for Erdos-Renyi matmul satisfies (in expectation)

#Words_moved = Ω(min( dn/P1/2 , d2n/P ) )– Contrast general lower bound: #Words_moved = Ω(d2n/(PM1/2)))

– Attained by divide-and-conquer algorithm that splits matrices along dimensions most likely to minimize cost

Page 29: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Other Results / Ongoing Work• Lots more work on

– Algorithms:• TSQR with Householder reconstruction • BLAS, LDLT, QR with pivoting, other pivoting schemes, eigenproblems, … • Both 2D (c=1) and 2.5D (c>1) • But only bandwidth may decrease with c>1, not latency• Strassen-like algorithms

– Platforms: • Multicore, cluster, GPU, cloud, heterogeneous, low-energy, …

• Integration into applications – ASPIRE Lab – computer vision

• video background subtraction, optical flow, …– AMPLab – building library using Spark for data analysis– CTF (with ANL): symmetric tensor contractions, on BG/Q

Page 30: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods• Architectural Implications

Page 31: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Recall optimal sequential Matmul

• Naïve code for i=1:n, for j=1:n, for k=1:n, C(i,j)+=A(i,k)*B(k,j)

• “Blocked” code: ... write A as n/b-by-n/b matrix of b-by-b blocks A[i,j] … ditto for B, C for i = 1:n/b, for j = 1:n/b, for k = 1:n/b, C[i,j] += A[i,k] * B[k,j] … b-by-b matrix multiply • Thm: Picking b = M1/2 attains lower bound: #words_moved = Ω(n3/M1/2)• Where does 1/2 come from?

Page 32: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

New Thm applied to Matmul• for i=1:n, for j=1:n, for k=1:n, C(i,j) += A(i,k)*B(k,j)• Record array indices in matrix Δ

• Solve LP for x = [xi,xj,xk]T: max 1Tx s.t. Δ x ≤ 1– Result: x = [1/2, 1/2, 1/2]T, 1Tx = 3/2 = sHBL

• Thm: #words_moved = Ω(n3/MSHBL-1)= Ω(n3/M1/2) Attained by block sizes Mxi,Mxj,Mxk = M1/2,M1/2,M1/2

i j k1 0 1 A

Δ = 0 1 1 B1 1 0 C

Page 33: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

New Thm applied to Direct N-Body• for i=1:n, for j=1:n, F(i) += force( P(i) , P(j) )• Record array indices in matrix Δ

• Solve LP for x = [xi,xj]T: max 1Tx s.t. Δ x ≤ 1– Result: x = [1,1], 1Tx = 2 = sHBL

• Thm: #words_moved = Ω(n2/MSHBL-1)= Ω(n2/M1) Attained by block sizes Mxi,Mxj = M1,M1

i j1 0 F

Δ = 1 0 P(i)0 1 P(j)

Page 34: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

N-Body Speedups on IBM-BG/P (Intrepid)8K cores, 32K particles

11.8x speedup

K. Yelick, E. Georganas, M. Driscoll, P. Koanantakool, E. Solomonik

Page 35: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

04/07/2015 CS267 Lecture 21

Some Applications

• Gravity, Turbulence, Molecular Dynamics, Plasma Simulation, Electron-Beam Lithography Simulation …

• Data Base Join• Hair ...

– www.fxguide.com/featured/brave-new-hair/– graphics.pixar.com/library/CurlyHairA/paper.pdf

35

Page 36: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

New Thm applied to Random Code• for i1=1:n, for i2=1:n, … , for i6=1:n A1(i1,i3,i6) += func1(A2(i1,i2,i4),A3(i2,i3,i5),A4(i3,i4,i6)) A5(i2,i6) += func2(A6(i1,i4,i5),A3(i3,i4,i6))• Record array indices in matrix Δ

• Solve LP for x = [x1,…,x7]T: max 1Tx s.t. Δ x ≤ 1– Result: x = [2/7,3/7,1/7,2/7,3/7,4/7], 1Tx = 15/7 = sHBL

• Thm: #words_moved = Ω(n6/MSHBL-1)= Ω(n6/M8/7) Attained by block sizes M2/7,M3/7,M1/7,M2/7,M3/7,M4/7

i1 i2 i3 i4 i5 i61 0 1 0 0 1 A1

1 1 0 1 0 0 A2

Δ = 0 1 1 0 1 0 A3

0 0 1 1 0 1 A3,A4

0 0 1 1 0 1 A5

1 0 0 1 1 0 A6

Page 37: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Approach to generalizing lower bounds• Matmul for i=1:n, for j=1:n, for k=1:n, C(i,j)+=A(i,k)*B(k,j) => for (i,j,k) in S = subset of Z3

Access locations indexed by (i,j), (i,k), (k,j)• General case for i1=1:n, for i2 = i1:m, … for ik = i3:i4 C(i1+2*i3-i7) = func(A(i2+3*i4,i1,i2,i1+i2,…),B(pnt(3*i4)),…) D(something else) = func(something else), … => for (i1,i2,…,ik) in S = subset of Zk

Access locations indexed by “projections”, eg φC (i1,i2,…,ik) = (i1+2*i3-i7)

φA (i1,i2,…,ik) = (i2+3*i4,i1,i2,i1+i2,…), …• Goal: Communication lower bounds, optimal algorithms for any

program that looks like this:– Thm: #words_moved = Ω(|S|/Me) for some constant e

Page 38: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

General Communication Bound

• Thm: Given a program with array refs given by projections φj, there is a linear program (LP) whose solution sHBL yields the lower bound

#words_moved = Ω (#iterations/MsHBL-1)• Proof uses recent result of Bennett/Carbery/Christ/Tao

generalizing inequalities of Cauchy-Schwartz, Hölder, Brascamp-Lieb, and Loomis-Whitney – Loomis-Whitney used by Irony/Tiskin/Toledo for matmul lower

bound

• Given S subset of Zk, group homomorphisms φ1, φ2, …, bound |S| in terms of |φ1(S)|, |φ2(S)|, … , |φm(S)|

• Thm (Christ/Tao/Carbery/Bennett): Given s1,…,sm

|S| ≤ Πj |φj(S)|sj

Page 39: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Is this bound attainable? (1/2)• But first: Can we write it down?

– Original theorem: Infinitely many inequalities in “LP”, but only finitely many different ones

– Thm: (bad news) Writing down all inequalities equivalent to Hilbert’s 10th problem over Q• conjectured to be undecidable

– Thm: (good news) Can decidably write down a subset of the constraints with the same solution sHBL

– Thm: (better news) Can write it down explicitly in many cases of interest• Ex: when all φj = {subset of indices}

– Thm: (good news) Easy to approximate• If you miss a constraint, the lower bound may be too large (i.e. sHBL too small) but still worth trying to attain, because

your algorithm will still communicate less

• Tarski-decidable to get superset of constraints (may get . sHBL too large)

Page 40: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Is this bound attainable? (2/2)

• Depends on loop dependencies• Best case: none, or reductions (matmul)• Thm: When all φj = {subset of indices}, dual of HBL-

LP gives optimal tile sizes: HBL-LP: minimize 1T*s s.t. sT*Δ ≥ 1T

Dual-HBL-LP: maximize 1T*x s.t. Δ*x ≤ 1 Then for sequential algorithm, tile ij by Mxj

• Ex: Matmul: s = [ 1/2 , 1/2 , 1/2 ]T = x• Extends to unimodular transforms of indices

Page 41: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Ongoing Work

• Make derivation of lower bounds efficient– Not just decidable

• Conjecture: Always attainable, modulo loop carried dependencies

• Open Question: How to incorporate dependencies into lower bounds?

• Integrate into compilers

Page 42: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural Implications

Page 43: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

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– Challenges: Poor partitioning, Preconditioning, Num. Stability

43

Page 44: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3• Works for any “well-partitioned” A

Page 45: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3

Page 46: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3

Page 47: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3

Page 48: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3

Page 49: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]

• Example: A tridiagonal, n=32, k=3

Page 50: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]• Sequential Algorithm

• Example: A tridiagonal, n=32, k=3

Step 1

Page 51: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]• Sequential Algorithm

• Example: A tridiagonal, n=32, k=3

Step 1 Step 2

Page 52: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]• Sequential Algorithm

• Example: A tridiagonal, n=32, k=3

Step 1 Step 2 Step 3

Page 53: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx] • Sequential Algorithm

• Example: A tridiagonal, n=32, k=3

Step 1 Step 2 Step 3 Step 4

Page 54: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]• Parallel Algorithm

• Example: A tridiagonal, n=32, k=3• Each processor communicates once with neighbors

Proc 1 Proc 2 Proc 3 Proc 4

Page 55: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

1 2 3 4 … … 32

x

A·x

A2·x

A3·x

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

• Replace k iterations of y = Ax with [Ax, A2x, …, Akx]• Parallel Algorithm

• Example: A tridiagonal, n=32, k=3• Each processor works on (overlapping) trapezoid

Proc 1 Proc 2 Proc 3 Proc 4

Page 56: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Same idea works for general sparse matrices

Communication Avoiding Kernels:The Matrix Powers Kernel : [Ax, A2x, …, Akx]

Simple block-row partitioning (hyper)graph partitioning

Top-to-bottom processing Traveling Salesman Problem

Page 57: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Minimizing Communication of GMRES to solve Ax=b

• GMRES: find x in span{b,Ab,…,Akb} 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, A2v, … , Akv ] [Q,R] = TSQR(W) … “Tall Skinny QR” build H from R solve LSQ problem with H

•Oops – W from power method, precision lost!57

Sequential case: #words moved decreases by a factor of kParallel case: #messages decreases by a factor of k

Page 58: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

“Monomial” basis [Ax,…,Akx] fails to converge

Different polynomial basis [p1(A)x,…,pk(A)x] does converge

58

Page 59: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Speed ups of GMRES on 8-core Intel Clovertown

[MHDY09]

59

Requires Co-tuning Kernels

Page 60: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

60

CA-BiCGStab

Page 61: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Sample Application Speedups

61

• Geometric Multigrid (GMG) w CA Bottom Solver• Compared BICGSTAB vs. CA-BICGSTAB with s = 4• Hopper at NERSC (Cray XE6), weak scaling: Up to

4096 MPI processes (24,576 cores total)

• Speedups for miniGMG benchmark (HPGMG benchmark predecessor) – 4.2x in bottom solve, 2.5x overall GMG solve

• Implemented as a solver option in BoxLib and CHOMBO AMR frameworks

– 3D LMC (a low-mach number combustion code)• 2.5x in bottom solve, 1.5x overall GMG solve

– 3D Nyx (an N-body and gas dynamics code)• 2x in bottom solve, 1.15x overall GMG solve

• Solve Horn-Schunck Optical Flow Equations • Compared CG vs. CA-CG with s = 3, 43% faster on NVIDIA GT 640 GPU

Page 62: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Summary of Iterative Linear Algebra

• New lower bounds, optimal algorithms, big speedups in theory and practice

• Lots of other progress, open problems– Many different algorithms reorganized

• More underway, more to be done

– Need to recognize stable variants more easily– Preconditioning

• Hierarchically Semiseparable Matrices, …

– Autotuning and synthesis• Different kinds of “sparse matrices”

Page 63: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Outline

• Survey state of the art of CA (Comm-Avoiding) algorithms– TSQR: Tall-Skinny QR– CA O(n3) 2.5D Matmul – Sparse Matrices

• Beyond linear algebra– Extending lower bounds to any algorithm with arrays– Communication-optimal N-body algorithm

• CA-Krylov methods • Architectural Implications

Page 64: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Architectural Implications

• Heterogeneous processors – What happens when communication costs, memory sizes,

etc. vary across machine?• Network contention

– What network topologies are needed to attain lower bounds?• Write-Avoiding algorithms

– What happens if writes are much more expensive than reads?

• Reproducible floating point summation– Can I get the same answer from run to run, even if

#processors change, etc?

Page 65: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

For more details

• Bebop.cs.berkeley.edu– 155 page survey in Acta Numerica

• CS267 – Berkeley’s Parallel Computing Course– Live broadcast in Spring 2015

• www.cs.berkeley.edu/~demmel• All slides, video available

– Prerecorded version broadcast since Spring 2013• www.xsede.org• Free supercomputer accounts to do homework• Free autograding of homework

Page 66: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Collaborators and Supporters• James Demmel, Kathy Yelick, Erin Carson, Orianna DeMasi, Aditya Devarakonda, David

Dinh, Michael Driscoll, Evangelos Georganas, Nicholas Knight, Penporn Koanantakool, Rebecca Roelofs, Yang You

• Michael Anderson, Grey Ballard, Austin Benson, Maryam Dehnavi, David Eliahu, Andrew Gearhart, Mark Hoemmen, Shoaib Kamil, Ben Lipshitz, Marghoob Mohiyuddin, Oded Schwartz, Edgar Solomonik, Omer Spillinger

• Aydin Buluc, Michael Christ, Alex Druinsky, Armando Fox, Laura Grigori, Ming Gu, Olga Holtz, Mathias Jacquelin, Kurt Keutzer, Amal Khabou, Sophie Moufawad, Tom Scanlon, Harsha Simhadri, Sam Williams

• Dulcenia Becker, Abhinav Bhatele, Sebastien Cayrols, Simplice Donfack, Jack Dongarra, Ioana Dumitriu, David Gleich, Jeff Hammond, Mike Heroux, Julien Langou, Devin Matthews, Michelle Strout, Mikolaj Szydlarksi, Sivan Toledo, Hua Xiang, Ichitaro Yamazaki, Inon Peled

• Members of ParLab, ASPIRE, BEBOP, CACHE, EASI, FASTMath, MAGMA, PLASMA• Thanks to DARPA, DOE, NSF, UC Discovery, INRIA, Intel, Microsoft, Mathworks, National

Instruments, NEC, Nokia, NVIDIA, Samsung, Oracle • bebop.cs.berkeley.edu

Page 67: Communication-Avoiding Algorithms for Linear Algebra and Beyond Jim Demmel EECS & Math Departments UC Berkeley bebop.cs.berkeley.edu

Summary

Don’t Communic…

67

Time to redesign all linear algebra, n-body, … algorithms and software

(and compilers)