[原创]利用Java多线程计算矩阵乘法
前言
前段时间在一本操作系统书籍上,看到了可以利用多线程来计算矩阵乘法的思想。例如下图中,A矩阵和B矩阵相乘得到C矩阵,那么A矩阵的每一行和B矩阵的每一列的相乘和加和,都可以交给一个线程来计算,最终得到cij这个元素。A矩阵维度是m*s,B矩阵是s*n,那么这个计算需要m*n个线程的参与,它是否一定比串行计算快呢?本文使用Java多线程一探究竟。
串行计算
矩阵乘法的串行计算方法是不难想到的。三层循环遍历计算即可。从外到内分别遍历A矩阵的行、B矩阵的列、A矩阵的列(即为B矩阵的行)即可。在计算的开始和结束时刻分别获取系统当前时刻,最后可得计算时间。以下是简单的代码片段。
// 串行验证
startTime = System.currentTimeMillis();
for (int i = 0; i < A.length; i++) {
for (int j = 0; j < B[0].length; j++) {
for (int k = 0; k < A[0].length; k++)
serial_result[i][j] += A[i][k] * B[k][j];
}
}
endTime = System.currentTimeMillis();
System.out.p