← Questions

Multiply two matrices across two GPUs

0
GoogleML designLast asked Sep 2025

Matrices A and B must be multiplied using two GPUs, with both matrices split between them. How do you partition the work (by rows, by columns, by blocks), what has to move between the GPUs, and how do you gather the result? Compare the communication cost of each scheme.

Your canvas
Draw your answer hereBoxes, arrows and notes, saved for you to come back to.Sign in to start drawing