pw_toy_code/pw_methods_driver.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
PROGRAM pw_methods_driver
  USE pw_methods_acc
  USE pw_methods_omp
  USE pw_methods_serial
  IMPLICIT NONE

  COMPLEX(8), DIMENSION(:), ALLOCATABLE :: array1, array2
  INTEGER, PARAMETER :: array_size = 2**20
  INTEGER :: idx
  INTEGER, DIMENSION(:), ALLOCATABLE :: gidx
  REAL(8) :: time_start, time_end

  ALLOCATE(array1(array_size), array2(2*array_size), gidx(array_size))

  DO idx = 1, array_size
    array1(idx) = CMPLX(idx, KIND=8)
    array2(idx) = CMPLX(-idx, KIND=8)
    array2(idx+array_size) = 0.0_8
    gidx(idx) = 2*idx
  END DO

  CALL CPU_TIME(time_start)
  CALL pw_axpy_ref_serial(array1, array2, gidx, 1.0_8)
  CALL CPU_TIME(time_end)

  WRITE(*,*) "SUM serial", SUM(array2), time_end-time_start

  DO idx = 1, array_size
    array1(idx) = CMPLX(idx, KIND=8)
    array2(idx) = CMPLX(-idx, KIND=8)
    array2(idx+array_size) = 0.0_8
    gidx(idx) = 2*idx
  END DO

  CALL CPU_TIME(time_start)
  CALL pw_axpy_ref_omp(array1, array2, gidx, 1.0_8)
  CALL CPU_TIME(time_end)

  WRITE(*,*) "SUM omp", SUM(array2), time_end-time_start

  DO idx = 1, array_size
    array1(idx) = CMPLX(idx, KIND=8)
    array2(idx) = CMPLX(-idx, KIND=8)
    array2(idx+array_size) = 0.0_8
    gidx(idx) = 2*idx
  END DO

  CALL CPU_TIME(time_start)
  CALL pw_axpy_ref_acc(array1, array2, gidx, 1.0_8)
  CALL CPU_TIME(time_end)

  WRITE(*,*) "SUM acc", SUM(array2), time_end-time_start

END PROGRAM pw_methods_driver