首页 > itarticle > fortran for fun 并行之mpi

fortran for fun 并行之mpi

admin 11月 13, 2020 0

MPI是常用的并行编程库，可以安装openmpi或mpich来获得mpi的API。

program learn_mpi
  use mpi
  implicit none
  integer :: ip, np, ctx, info
  real :: wtime
  call mpi_init(info)
  call mpi_comm_size(MPI_COMM_WORLD, np, info)
  call mpi_comm_rank(MPI_COMM_WORLD, ip, info)
  if(ip == 0) then
    print*, 'number of process = ',np
    wtime = mpi_wtime()
  endif
  print*, 'hello mpi from process ',ip
  if(ip==0) then
    wtime = mpi_wtime() - wtime
    print*, 'Time elapsed = ',wtime,' seconds.'
  endif
  call mpi_finalize(info)
end program learn_mpi

结果

执行mpirun -n 4 learn_mpi得到

number of process =            4
hello mpi from process            0
hello mpi from process            1
hello mpi from process            2
hello mpi from process            3
Time elapsed =    4.0020095184445381E-006  seconds.