1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
!
! Copyright (C) by Argonne National Laboratory
! See COPYRIGHT in top-level directory
!
program main
use mpi_f08
implicit none
integer :: size, rank, ierr, errs
logical :: verbose
common /flags/ verbose
character (len=10) :: name='test9'
character (len=80) :: title='test 9: Send/Recv 2d array column slice - iar_2d(1:7:3,2:6:2)'
integer :: i, j
integer, dimension(9,9) :: iar_2d, iar_2dch
type(MPI_Status) status
errs = 0
verbose = .false.
call MTest_Init(ierr)
call mpi_comm_size(MPI_COMM_WORLD, size, ierr)
call mpi_comm_rank(MPI_COMM_WORLD, rank, ierr)
if (rank .eq. 0) then
if (verbose) print *, ' ===== ', trim(title), ' ====='
do i=1,9
do j=1,9
iar_2d(j,i) = (j * 10) + i
end do
end do
do i=1,9
if (verbose) print *, (iar_2d(i,j),j=1,9)
end do
else
do i=1,9
do j=1,9
iar_2d(j,i) = 0
iar_2dch(j,i) = 0
end do
end do
do i=2,6,2
do j=1,7,3
iar_2dch(j,i) = (j * 10) + i
end do
end do
endif
if (rank .eq. 0) then
call mpi_send(iar_2d(1:7:3,2:6:2), 9, MPI_INTEGER, 1, 123, MPI_COMM_WORLD, ierr);
if (ierr .ne. MPI_SUCCESS) then
print *,"PE ",rank,": ",name,": mpi_send exited in error (",ierr,")"
errs = errs + 1
endif
else if (rank .eq. 1) then
call mpi_recv(iar_2d(1:7:3,2:6:2), 9, MPI_INTEGER, 0, 123, MPI_COMM_WORLD, status, ierr);
if (ierr .ne. MPI_SUCCESS) then
print *,"PE ",rank,": ",name,"mpi_recv exited in error (",ierr,")"
errs = errs + 1
endif
do i=2,6,2
do j=1,7,3
if (iar_2d(j,i) .ne. iar_2dch(j,i)) then
print *, "rank 1: iar_2d(", j, ",", i, ")=", iar_2d(j,i), ", expected ", iar_2dch(j,i)
errs = errs + 1
endif
end do
end do
if (errs .eq. 0) then
if (verbose) print *, "PE ", rank,": PASS - ", trim(title)
else
print *, "PE ", rank,": FAIL - ", trim(title)
endif
do i=1,9
if (verbose) print *, (iar_2d(i,j),j=1,9)
end do
endif
call MTest_Finalize(errs)
end program
|
