陈翊辉
SA19011116
[toc]蝶形运算是并行计算中一种基本的方法,快速傅里叶变换就通常使用蝶形运算,而蝶形求和也是最简单的蝶形计算,可以使$2^n$个处理器经过$n$步得到全和。其并行算法如下:
输入:n个数,每个处理器上各有一个,n是2的幂,n=2^M
输出:n个数的和,每个处理器上都有
Begin
for k=1 to m do
计算对应处理器号t
按蝶形计算向对应处理器t发送自己的部分和(第一次是初始数)
按蝶形计算从对应处理器t接收部分和并加到自己部分和(第一次是初始数)
end for
End
其中对应处理器号t
计算
base = 2 ** k;
group = rank / base;
offset = rank % base;
target = (group % 2 ? group - 1 : group + 1) * base + offset;
在蝶形运算中,共有$log_2n$步,每一步需要每个处理器给对应处理器发送,并从对应处理器接收,如果都先发或先收则必定造成死锁,这里用简单的rank和target比较判断,也可以考虑异步收发或者用MPI_Sendrecv
。
for (int i = 0; i < steps; ++i) {
int group = rank / base;
int offset = rank % base;
int target = (group % 2 ? group - 1 : group + 1) * base + offset;
if (rank < target) {
MPI_Send(reinterpret_cast<void*>(&send), 1, MPI_INT, target, target, MPI_COMM_WORLD);
MPI_Recv(reinterpret_cast<void*>(&recv), 1, MPI_INT, target, rank, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
} else {
MPI_Recv(reinterpret_cast<void*>(&recv), 1, MPI_INT, target, rank, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
MPI_Send(reinterpret_cast<void*>(&send), 1, MPI_INT, target, target, MPI_COMM_WORLD);
}
send += recv;
base *= 2;
}
完整代码见附件
二叉树求和使$2^n$个处理器经过$2n$步得到全和,前$n$步是树上的归约求和过程,后$n$步是发送结果过程。其并行算法如下:
输入:n个数,每个处理器上各有一个,n是2的幂,n=2^M
输出:n个数的和,每个处理器上都有
Begin
for k=1 to m do
叶子结点向根节点发送部分和(第一次是初始数)
根结点求和,根结点继续下一轮,非根结点结束
end for
for k=1 to m do
根结点向两个孩子结点发送全和
end for
End
二叉树根的选取可以任意,一种比较简单的二叉树构建方法是按处理器编号0,1;2,3;...;构建,每次编号小的作为根结点,然后进行下一轮运算。
一种比较简单的确定二叉树方法是按处理器编号0,1;2,3;...;构建,每次编号小的作为根结点,然后进行下一轮运算。比如开始时,所有结点都参与,即mod1=0的参与,mod2=0的作为根结点;第2步,mod2=0的参与,mod4=0的作为根结点......发送结果时反过来。
根结点和非根节点一个收,一个发,只需要用最简单的MPI_Send
和MPI_Recv
即可。
int base = 1;
for (int i = 0; i < steps; ++i) {
if (rank % base == 0) {
int b = base * 2;
if (rank % b == 0) { // recv
MPI_Recv(&recv, 1, MPI_INT, rank + base, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
} else { // send
MPI_Send(&send, 1, MPI_INT, rank - base, 0, MPI_COMM_WORLD);
}
}
send += recv;
base *= 2;
}
base /= 2;
for (int i = 0; i < steps; ++i) {
if (rank % base == 0) {
int b = base * 2;
if (rank % b == 0) { // recv
MPI_Send(&send, 1, MPI_INT, rank + base, 0, MPI_COMM_WORLD);
} else { // send
MPI_Recv(&send, 1, MPI_INT, rank - base, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
}
}
base /= 2;
}
完整代码见附件
和其他并行矩阵乘法类似,FOX矩阵乘法基本原理也是矩阵分块乘法,不同的是使用了不同的矩阵块传送策略。
输入:n*n矩阵A,n*n矩阵B,初始时A和B分成p个子矩阵,处理器P[i,j]存有块A[i,j]和B[i,j](q * q = p,处理器编号0,0;0,1;...;q-1,q-1)
输出:n*n矩阵C,每个处理器P[i,j]存有块C[i,j]
Begin
for k = 1 to q do
处理器P[i,(i+k) mod q]向所在行广播其A子块
各处理器将接收到的A子块和自己的B子块进行矩阵乘,并加到结果部分和
B子块向上循环移动
end for
End
MPI实现主要有几个部分:
其中读取,输出,部分矩阵乘虽然是比较基础的操作,但考虑到有广播分发,和接收的操作,需要考虑其数据摆放(layout)。
输入数据矩阵A和B按正常顺序放在文本文件中。
4
0 1 2 3
4 5 6 7
8 9 a b
c d e f
如果直接按顺序读入内存,则在散播子块时比较不方便,因为每一子块的内存不是连续的,这样要么需要再申请一块内存搬动数据,要么就需要分部分传输每个子块。因而考虑读入时就按子块连续的方式存放。
比如4*4矩阵,分到4个处理器,读入数据摆放如下,这样直接使用一个MPI_Scatter
就可完成子块分发。
0 1 4 5
2 3 6 7
8 9 c d
a b e f
在运算前,需要将处理器按行和列分到不同的通信域中,因为处理器初始的编号是一维的,而算法需要二维的编号,并且一些在行内,列内的操作也需要各自的通信域。
运算过程中,广播A子块使用MPI_Bcast
,在行通信域内广播
B子块的循环移动,可以使用MPI_Sendrecv
,每个处理器的源是下一行,目标是上一行(环形考虑)
矩阵C的接收和矩阵AB的分发正好相反,使用MPI_Gather
汇集到0处理器上,这时接收到的C矩阵不是正常顺序,需要根据一定的顺序输出。
数据摆放变换,矩阵乘法
class MatWrap {
private:
float* data_;
int n_;
public:
MatWrap() {}
MatWrap(float* data, int n) : data_(data), n_(n) {}
float* operator[](size_t n) const { return data_ + n * n_; }
float* split_map(int sqrt_q, int i, int j) const {
int sub_n = n_ / sqrt_q;
return data_ + (i / sub_n * sqrt_q + j / sub_n) * sub_n * sub_n +
i % sub_n * sub_n + j % sub_n;
}
friend void MatMultAdd(const MatWrap& a, const MatWrap& b, MatWrap& c);
};
void MatMultAdd(const MatWrap& a, const MatWrap& b, MatWrap& c) {
assert(a.n_ == b.n_);
assert(a.n_ == c.n_);
int n = a.n_;
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
for (int k = 0; k < n; ++k) {
c[i][j] += a[i][k] * b[k][j];
}
}
}
}
完整代码见附件
并行计算部分
// broadcast sub matrix
MPI_Scatter(mat_a, sub_n * sub_n, MPI_FLOAT, sub_mat_a, sub_n * sub_n,
MPI_FLOAT, 0, MPI_COMM_WORLD);
MPI_Scatter(mat_b, sub_n * sub_n, MPI_FLOAT, sub_mat_b, sub_n * sub_n,
MPI_FLOAT, 0, MPI_COMM_WORLD);
// split comm in col and row
MPI_Comm col_world, row_world;
int col_rank = rank % sqrt_q;
int row_rank = rank / sqrt_q;
MPI_Comm_split(MPI_COMM_WORLD, col_rank, row_rank, &col_world);
MPI_Comm_split(MPI_COMM_WORLD, row_rank, col_rank, &row_world);
// compute
for (int i = 0; i < sqrt_q; ++i) {
// broadcast sub_a
int send_root = (row_rank + i) % sqrt_q;
if (col_rank == (row_rank + i) % sqrt_q) {
memcpy(sub_mat_comm, sub_mat_a, sub_n * sub_n * sizeof(float));
}
MPI_Bcast(sub_mat_comm, sub_n * sub_n, MPI_FLOAT, send_root, row_world);
// calculate sub mat gemm
MatMultAdd(sub_comm, sub_b, sub_c);
// swap sub_b
MPI_Sendrecv_replace(
sub_mat_b, sub_n * sub_n, MPI_FLOAT, (row_rank + sqrt_q - 1) % sqrt_q,
1, (row_rank + 1) % sqrt_q, 1, col_world, MPI_STATUS_IGNORE);
}
// gather result
MPI_Gather(sub_mat_c, sub_n * sub_n, MPI_FLOAT, mat_c, sub_n * sub_n,
MPI_FLOAT, 0, MPI_COMM_WORLD);
设系统中总计有N个进程,其中P个进程作为参数服务器进程,而Q个进程作为工作进程(N = P + Q, 且 0 < P << Q)。工作进程和服务器进程的互动过程如下:
每个参数服务器发送该平均值给它对应的所有工作进程,然后再等待……
do while true
工作进程i产生随机数
i mod p参数服务器接收进程i的随机数
所有服务器通信计算所有数据平均值
参数服务器发送平均值给工作进程
end do
为了方便各种通信,需要划分通信域:
划分好通信域之后就比较简单了,
因为最终只需要计算一个平均值,并且要等待每个工作进程,直接使用MPI的归约操作
MPI_Reduce
,其中的操作MPI_SUM
,这样每个服务器都有着负责的工作进程的数字的和。
所有服务器通信计算平均值
这里需要每个服务器都获得所有工作进程的平均,而现在每个服务器只有自己负责工作进程的部分和,考虑使用MPI_Allreduce
,其中的操作仍为MPI_SUM
,这时每个服务器都有所有工作进程的和,再除以工作进程数即可获得进程的平均数。
服务器发送平均值给工作进程 每个服务器把平均数广播给工作进程即可。
// server or client
int SorC = rank >= SERVER_NUM;
MPI_Comm serverClient, service;
MPI_Comm_split(MPI_COMM_WORLD, SorC, -1, &serverClient);
int service_num = rank % SERVER_NUM;
MPI_Comm_split(MPI_COMM_WORLD, service_num, SorC, &service);
int service_rank;
MPI_Comm_rank(service, &service_rank);
// start
float value;
float sum;
for (int i = 0; i < LOOP; ++i) {
if (SorC) { // client
value = dis(gen);
// printf("c:%d, %d, %f\n", rank, service_rank, value);
} else { // server
value = 0;
}
MPI_Reduce(&value, &sum, 1, MPI_FLOAT, MPI_SUM, 0, service);
if (!SorC) {
// printf("s:%d, %d, %f\n", rank, service_rank, sum);
MPI_Allreduce(&sum, &value, 1, MPI_FLOAT, MPI_SUM, serverClient);
value /= (size - SERVER_NUM);
// printf("s:%d, %d, %f\n", rank, service_rank, value);
}
MPI_Bcast(&value, 1, MPI_FLOAT, 0, service);
// printf("c:%d, %f\n", rank, value);
}
完整代码见附件
长短期记忆(英语:Long Short-Term Memory,LSTM)是一种时间递归神经网络(RNN),论文首次发表于1997年。由于独特的设计结构,LSTM适合于处理和预测时间序列中间隔和延迟非常长的重要事件。LSTM可以应用在语音识别,机器翻译,自然语言处理,手写识别等机器学习应用。
$$ it=\text{sigmoid}(W{ii}xt+b{ii}+W{hi}h{(t-1)}+b_{hi})\ ft=\text{sigmoid}(W{if}xt+b{if}+W{hf}h{(t-1)}+b_{hf})\ gt=\text{tanh}(W{ig}xt+b{ig}+W{hc}h{(t-1)}+b_{hg})\ ot=\text{sigmoid}(W{io}xt+b{io}+W{ho}h{(t-1)}+b_{ho})\ c_t=ft\cdot c{(t-1)}+i_t\cdot g_t\ h_t=o_t\cdot \text{tanh}(c_t) $$
从上面公式来看,输出$c_t,ht$依赖于$c{(t-1)},f_t,g_t,i_t,o_t$,而$i_t,f_t,g_t,ot$又依赖于$h{(t-1)}$,对于这种随时间的迭代计算,不同时间$t$之间不能并行计算,因而考虑$i_t,f_t,g_t,o_t$可以并行计算,而在$i_t,f_t,g_t,o_t$内有矩阵乘加计算,也可以使用分块矩阵的并行计算。
对于上面公式的矩阵乘加计算,使用增广矩阵形式,化为矩阵乘。 $$ it=\text{sigmoid}(W{ii}xt+W{hi}h_{(t-1)})\ ft=\text{sigmoid}(W{if}xt+W{hf}h_{(t-1)})\ gt=\text{tanh}(W{ig}xt+W{hc}h_{(t-1)})\ ot=\text{sigmoid}(W{io}xt+W{ho}h_{(t-1)})\ c_t=ft\cdot c{(t-1)}+i_t\cdot g_t\ h_t=o_t\cdot \text{tanh}(c_t) $$ 对于LSTM的各个时间点的计算,使用MPI分配在不同的节点上,使用流水化设计,节点间只需传输$h_t,c_t$
对于LSTM的一个时间点内的计算,考虑在同一个节点上使用MPI分为$i_t,f_t,g_t,o_t$4个部分计算,需要广播$x_t$
$i_t,f_t,g_t,o_t$每个部分内的矩阵乘法使用OpenMP计算,OpenMP的线程数设置为,一个节点上的处理器个数/4
章 | 15 | 19 | 22 |
---|---|---|---|
算法程序数 | 4 | 2 | 2 |
分到的程序号 | 1 | 1 | 1 |
分到的程序 | closure | gauss | fft |