如何初始化两个不同的blacs上下文？

Question

我有一台带有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通信器吗？

Answer 1

作为一个答案和例子，我分享这个实现：

    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_get (第三个参数)在一个新变量中接收。

在本例中，我发现的是，上下文的值似乎遵循某种求和规则，因此在初始化1x1网格和再次初始化4x4网格之后，4x4网格句柄的值在所有进程中都不一样。