我正在使用使用 ScaLAPACK 的 Intel-MKL 执行 Cholesky 分解。我根据这个例子分发了矩阵,其中矩阵分布在大小相等的块中(即 Nb x Mb)。我试图让每个块都有自己的大小,具体取决于它属于哪个进程,这样我就可以进行更多实验,也许会获得更好的性能。
检查这个问题,以便更好地理解我在说什么。我不会发布我的代码,因为它太大了(是的,小例子也太大了,我检查了)并且分发似乎运行良好。但是,ScaLAPACK 似乎假设矩阵分布在大小相等的块中?
例如,我正在使用这个:
int nrows = numroc_(&N, &Nb, &myrow, &iZERO, &procrows);
int ncols = numroc_(&M, &Mb, &mycol, &iZERO, &proccols);
其中(取自手册):
NB(全局输入)整数块大小,分布式矩阵被分割成的块的大小。
那么,ScaLAPACK 是否允许具有不相等块大小的分布式矩阵?
如果我打印这样的信息,对于 8x8 矩阵:
std::cout << Nb << " " << Mb << " " << nrows << " " << ncols << " " << myid << std::endl;
我得到这个:
3 3 5 5 0
1 1 4 4 1
1 1 4 4 2
1 1 4 4 3
只需交换前两个块大小,即可:
1 1 4 4 0
3 3 5 3 1
1 1 4 4 2
1 1 4 4 3
这对于 8x8 矩阵没有意义。