MPI和OpenMP实现矩阵相乘
一、MPI实现矩阵相乘
-
首先要配置要MPI多节点集群环境,没有配置好的可以参考我的上一篇文章
-
原理:
-
A×B=CA\times B=CA×B=C ,这里矩阵 A,B,CA,B,CA,B,C 大小都为 N×NN \times NN×N ,代码中默认为128
-
main节点将矩阵 AAA 按行分成多段后,发给salve节点
-
main节点将矩阵 BBB 整体发给slave节点
-
salve节点接受main节点传过来的 A,BA,BA,B 然后利用下面公式计算 CijC_{ij}Cij 然后传回给main节点
-
Cij=∑k=1NAik×Bkj C_{ij} = \sum_{k=1}^{N}A_{ik}\times B_{kj} Cij=k=1∑NAik×Bkj
3. 运行:
- 编写一个profile文件,里面是关于节点的配置,仅main节点需要,内容如下
node01:2
node02:2
-
编写源代码文件gemm.cpp,两个虚拟机都需要
-
编译gemm.cpp文件,两个虚拟机都需要编译
mpic++ gemm.cpp -o gemm
- mpi


1万+

被折叠的 条评论
为什么被折叠?



