You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: examples/mlp-mpi/README.md
+7-1Lines changed: 7 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,4 +1,10 @@
1
-
# Computing a MLP sigmoid(A@B)@C on multiple ranks using MPI through MLIR
1
+
# Computing an MLP sigmoid(A@B)@C on multiple ranks using MPI through MLIR
2
+
3
+
This example shows how MLIR's sharding infrastructure can be used to distribute data and computation across multiple nodes with non-shared memory.
4
+
5
+
Currently, only the lower part of the sharding pipeline is used: `shard-partition`, `convert-shard-to-mpi`, and lowering to LLVM. Therefore, the ingress MLIR is fully annotated.
6
+
7
+
The example implements a "single MLP", following a 1D/2D weight-stationary partition strategy as described in figures 2a and 2b of https://arxiv.org/pdf/2211.05102.
0 commit comments