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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
!
! Copyright (C) 2013 Quantum ESPRESSO group
! This file is distributed under the terms of the
! GNU General Public License. See the file `License'
! in the root directory of the present distribution,
! or http://www.gnu.org/copyleft/gpl.txt .
!
!----------------------------------------------------------------------------
MODULE mp_global
!----------------------------------------------------------------------------
!
! ... Wrapper module, for compatibility. Contains a few "leftover" variables
! ... used for checks (all the *_file variables, read from data file),
! ... plus the routine mp_startup initializing MPI and the command line,
! ... plus the routine mp_global_end stopping MPI.
! ... Do not use this module to reference variables (e.g. communicators)
! ... belonging to each of the various parallelization levels:
! ... use the specific modules instead.
! ... PLEASE DO NOT ADD NEW STUFF TO THIS MODULE. Removing stuff is ok.
!
USE mp_world, ONLY: world_comm, mp_world_start, mp_world_end
USE mp_images
USE mp_pools
USE mp_bands
USE mp_bands_TDDFPT
USE mp_exx
USE mp_orthopools
!
IMPLICIT NONE
SAVE
!
! ... number of processors for the various groups: values read from file
!
INTEGER :: nproc_file = 1
INTEGER :: nproc_image_file = 1
INTEGER :: nproc_pool_file = 1
INTEGER :: nproc_ortho_file = 1
INTEGER :: nproc_bgrp_file = 1
INTEGER :: ntask_groups_file= 1
INTEGER :: nyfft_file= 1
!
CONTAINS
!
!-----------------------------------------------------------------------
SUBROUTINE mp_startup ( my_world_comm, start_images )
!-----------------------------------------------------------------------
! ... This wrapper subroutine initializes all parallelization levels.
! ... If option with_images=.true., processes are organized into images,
! ... each performing a quasi-indipendent calculation, such as a point
! .. in configuration space (NEB) or a phonon irrep (PHonon)
! ... Within each image processes are further subdivided into various
! ... groups and parallelization levels.
! ... IMPORTANT NOTICE 1: since the command line is read here, it may be
! ... convenient to call it in serial execution as well
! ... IMPORTANT NOTICE 2: most parallelization levels are initialized here
! ... but at least some will be moved to a later stage
!
USE command_line_options, ONLY : get_command_line, &
nimage_, npool_, nband_, ntg_, nyfft_
USE parallel_include
!
IMPLICIT NONE
INTEGER, INTENT(IN), OPTIONAL :: my_world_comm
LOGICAL, INTENT(IN), OPTIONAL :: start_images
LOGICAL :: do_images
INTEGER :: my_comm
!
my_comm = MPI_COMM_WORLD
IF ( PRESENT(my_world_comm) ) my_comm = my_world_comm
!
CALL mp_world_start( my_comm )
CALL get_command_line ( )
!
do_images = .FALSE.
IF ( PRESENT(start_images) ) do_images = start_images
IF ( do_images ) THEN
CALL mp_start_images ( nimage_, world_comm )
ELSE
CALL mp_init_image ( world_comm )
END IF
!
CALL mp_start_pools ( npool_, intra_image_comm )
! Init orthopools is done during EXX bootstrap but,
! if they become more used, do it here:
! CALL mp_start_orthopools ( intra_image_comm )
CALL mp_start_bands ( nband_, ntg_, nyfft_, intra_pool_comm )
CALL mp_start_exx ( nband_, ntg_, intra_pool_comm )
!
RETURN
!
END SUBROUTINE mp_startup
!
!-----------------------------------------------------------------------
SUBROUTINE mp_global_end ( )
!-----------------------------------------------------------------------
!
USE mp, ONLY : mp_comm_free
!
CALL mp_comm_free ( intra_bgrp_comm )
CALL mp_comm_free ( inter_bgrp_comm )
CALL mp_comm_free ( intra_pool_comm )
CALL mp_comm_free ( inter_pool_comm )
CALL mp_stop_orthopools( ) ! cleans orthopools if used in exx
CALL mp_world_end( )
!
RETURN
!
END SUBROUTINE mp_global_end
!
END MODULE mp_global
|
