pw_toy_code/pw_methods_omp.F at main · fstein93/pw_toy_code · GitHub

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
MODULE pw_methods_omp

      PRIVATE

      PUBLIC :: pw_axpy_ref_omp

      CONTAINS

             SUBROUTINE pw_axpy_ref_omp(array1, array2, gidx, alpha)
                     COMPLEX(KIND=8), CONTIGUOUS, DIMENSION(:), INTENT(IN) :: array1
                     COMPLEX(KIND=8), CONTIGUOUS, DIMENSION(:), INTENT(INOUT) :: array2
                     INTEGER, DIMENSION(:), CONTIGUOUS, INTENT(IN) :: gidx
                     REAL(KIND=8), INTENT(IN) :: alpha

                     INTEGER :: ng, ng1, ng2, i, j

                     ng1 = SIZE(array1)
                     ng2 = SIZE(array2)
                     ng = MIN(ng1, ng2)

                     IF (ng1 >= ng2) THEN
                        IF (alpha == 1.0_8) THEN
!$OMP PARALLEL DO PRIVATE(i,j) DEFAULT(NONE) SHARED(ng, gidx, array1, array2)
                           DO i = 1, ng
                              j = gidx(i)
                              array2(i) = array2(i) + array1(j)
                           END DO
!$OMP END PARALLEL DO
                        ELSE IF (alpha /= 0.0_8) THEN
!$OMP PARALLEL DO PRIVATE(i,j) DEFAULT(NONE) SHARED(alpha, ng, gidx, array1, array2)
                           DO i = 1, ng
                              j = gidx(i)
                              array2(i) = array2(i) + alpha*array1(j)
                           END DO
!$OMP END PARALLEL DO
                        END IF
                     ELSE
                        IF (alpha == 1.0_8) THEN
!$OMP PARALLEL DO PRIVATE(i,j) DEFAULT(NONE) SHARED(ng, gidx, array1, array2)
                           DO i = 1, ng
                              j = gidx(i)
                              array2(j) = array2(j) + array1(i)
                           END DO
!$OMP END PARALLEL DO
                        ELSE IF (alpha /= 0.0_8) THEN
!$OMP PARALLEL DO PRIVATE(i,j) DEFAULT(NONE) SHARED(alpha, ng, gidx, array1, array2)
                           DO i = 1, ng
                              j = gidx(i)
                              array2(j) = array2(j) + alpha*array1(i)
                           END DO
!$OMP END PARALLEL DO
                        END IF
                     END IF
                     END SUBROUTINE pw_axpy_ref_omp
END MODULE pw_methods_omp