Question

我有两个数组，数组global有8个值，它将在数组local中散布，有2个值。我试图做的是，采用大数组，分成小数组，做一些工作，然后把它重新组合起来。

问题：
即使我成功地分散了数据，所写的do循环仅适用于第一个子数组local。我想要的是分散的local数组中的所有整数应该乘以2，然后收集到global数组中。

do循环的代码（这里已完成一些工作）：

do j = 1,2
  local(j) = j*2
  print *, j
end do

这里是完整的代码。如果你往下走，你会注意到我需要你帮助的部分。

MODULE MPI
    IMPLICIT NONE
    INCLUDE   'mpif.h'
    INTEGER   :: MYID,TOTPS, IERR, MPISTTS

CONTAINS 

    SUBROUTINE MPIINIT
        IMPLICIT NONE
        CALL MPI_INIT( IERR ) 
        CALL MPI_COMM_RANK(MPI_COMM_WORLD,MYID,IERR)
        CALL MPI_COMM_SIZE(MPI_COMM_WORLD,TOTPS,IERR)
    RETURN
    END SUBROUTINE MPIINIT
END MODULE MPI

PROGRAM SCATTER 
    USE MPI
    IMPLICIT NONE 
    CALL MPIINIT
    CALL TEST
    CALL  MPI_FINALIZE(IERR)

CONTAINS 

SUBROUTINE TEST
    USE MPI 
    IMPLICIT NONE 
    INTEGER :: I,J
    INTEGER,DIMENSION(8) :: GLOBAL
    INTEGER,DIMENSION(2) :: LOCAL 

    if (myid .eq. 0) then 
    do i = 1,8
    global(i) = i
    end do 
    end if

    call mpi_scatter(global,2,mpi_integer,local,2,mpi_integer,0, &
    mpi_comm_world,ierr)
    print*,"task",myid,":",local
    call mpi_barrier(mpi_comm_world,ierr)

    !!!!!!! do some work here
    do j = 1,2
    local(j) = j*2
    print*,j
    end do
    !!!!!! end work 


    call mpi_gather(local,2,mpi_integer,global,2,mpi_integer,0, &
    mpi_comm_world,ierr)
    if(myid .eq. 0) then
    print*,"task",myid,":",global
    end if
END SUBROUTINE TEST
END PROGRAM SCATTER

注意：

（1）我一直在阅读＆amp;从this线程学习，但现在看起来很有挑战性（2）运行代码mpif90 SCATTER.f90 .. mpirun -np 4 ./a.out

输出：

    task 0 : 1 2
    task 1 : 3 4
    task 2 : 5 6
    task 3 : 7 8
    1
    2
    1
    2
    1
    2
    1
    2
   task 0 : 2 4 2 4 2 4 2 4

我想得到的是：task 0 : 2 4 6 8 10 12 14 16

Answer 1

你写了

local(j) = j * 2
print*, j

我认为这不符合你的想法。

你可能打算写

local(j) = local(j) * 2
print*, local(j)

在MPI SCATTER和GATHER中，循环卡在第一个子阵列上

1 个答案: