blob: f6b58f7c0779346287badc763423cb0277494f8d [file] [log] [blame]
 program teams2 use omp_lib integer :: i, j, err err = 0 !\$omp teams reduction(+:err) err = err + bar (0, 0, 0) !\$omp end teams if (err .ne. 0) stop 1 !\$omp teams reduction(+:err) err = err + bar (1, 0, 0) !\$omp end teams if (err .ne. 0) stop 2 !\$omp teams reduction(+:err) !\$omp distribute do i = 0, 63 err = err + bar (2, i, 0) end do !\$omp end teams if (err .ne. 0) stop 3 !\$omp teams reduction(+:err) !\$omp distribute do i = 0, 63 !\$omp parallel do reduction(+:err) do j = 0, 31 err = err + bar (3, i, j) end do end do !\$omp end teams if (err .ne. 0) stop 4 contains subroutine foo (x, y, z, a, b) integer :: x, y, z, a, b(64), i, j if (x .eq. 0) then do i = 0, 63 !\$omp parallel do shared (a, b) do j = 0, 31 call foo (3, i, j, a, b) end do end do else if (x .eq. 1) then !\$omp distribute dist_schedule (static, 1) do i = 0, 63 !\$omp parallel do shared (a, b) do j = 0, 31 call foo (3, i, j, a, b) end do end do else if (x .eq. 2) then !\$omp parallel do shared (a, b) do j = 0, 31 call foo (3, y, j, a, b) end do else !\$omp atomic b(y + 1) = b(y + 1) + z !\$omp end atomic !\$omp atomic a = a + 1 !\$omp end atomic end if end subroutine integer function bar (x, y, z) use omp_lib integer :: x, y, z, a, b(64), i, c, d, e, f a = 8 do i = 0, 63 b(i + 1) = i end do call foo (x, y, z, a, b) if (x .eq. 0) then if (a .ne. 8 + 64 * 32) then bar = 1 return end if do i = 0, 63 if (b(i + 1) .ne. i + 31 * 32 / 2) then bar = 1 return end if end do else if (x .eq. 1) then c = omp_get_num_teams () d = omp_get_team_num () e = d f = 0 do i = 0, 63 if (i .eq. e) then if (b(i + 1) .ne. i + 31 * 32 / 2) then bar = 1 return end if f = f + 1 e = e + c else if (b(i + 1) .ne. i) then bar = 1 return end if end do if (a .lt. 8 .or. a > 8 + f * 32) then bar = 1 return end if else if (x .eq. 2) then if (a .ne. 8 + 32) then bar = 1 return end if do i = 0, 63 if (i .eq. y) then c = 31 * 32 / 2 else c = 0 end if if (b(i + 1) .ne. i + c) then bar = 1 return end if end do else if (x .eq. 3) then if (a .ne. 8 + 1) then bar = 1 return end if do i = 0, 63 if (i .eq. y) then c = z else c = 0 end if if (b (i + 1) .ne. i + c) then bar = 1 return end if end do end if bar = 0 return end function end program