4

我有一台带有nproc处理器的计算机，我想初始化两个 blacs 网格，一个是维度p x q = nprocs，一个是维度1 x 1。

假设 MPI 已经初始化并且例程找到好的块大小，第一个网格通过以下方式初始化

call blacs_get(   -1, 0, self%context  )
call blacs_gridinit( self%context, 'R', self%nprows, self%npcols )
call blacs_gridinfo( self%context, self%nprows, self%npcols, self%myrow, self%mycol )

但是如何设置第二个？我必须先介绍另一个mpi通信器吗？

4

1 回答 1

0

作为答案和示例，我分享了这个实现：

    call blacs_get(   -1, 0, self%context  )
    call blacs_gridinit( self%context, 'R', self%nprows, self%npcols )
    call blacs_gridinfo( self%context, self%nprows, self%npcols, self%myrow, self%mycol )

    print*, "A ", self%context, self%nprows, self%npcols, self%myrow, self%mycol

    call sleep(1)

    call blacs_get(   -1, 0, val  )
    call blacs_gridinit( val, 'R', 1, 1 )
    call blacs_gridinfo( val, self%nprows, self%npcols, self%myrow, self%mycol )

    call sleep(1)

    print*, "B ", val, self%nprows, self%npcols, self%myrow, self%mycol

    call sleep(1)

    call blacs_get(   -1, 0, val2  )
    call blacs_gridinit( val2, 'R', 2, 2 )
    call blacs_gridinfo( val2, self%nprows, self%npcols, self%myrow, self%mycol )

    call sleep(1)
    print*, "C ", val2, self%nprows, self%npcols, self%myrow, self%mycol

这增加了三个 blacs 上下文，无需初始化另一个 MPI 通信器，在四个内核上相当于以下输出：

 A            0           2           2           1           1
 A            0           2           2           0           0
 A            0           2           2           1           0
 A            0           2           2           0           1




 B           -1          -1          -1          -1          -1
 B           -1          -1          -1          -1          -1
 B           -1          -1          -1          -1          -1
 B            1           1           1           0           0




 C            1           2           2           1           0
 C            1           2           2           1           1
 C            1           2           2           0           1
 C            2           2           2           0           0

所以，关键是 blacs_gridinit 的第一个参数是输入/输出参数，需要所有进程的 globale blacs context 作为输入。通过调用 blacs_get，它在一个新变量中被接收，第三个参数。

在这种情况下，我发现非常反直觉的是，上下文的值似乎遵循某种求和规则，因此在初始化 1x1 网格然后再次初始化 4x4 网格之后，4x4 网格句柄的值不是所有进程都一样。

于 2015-06-10T08:53:02.667 回答