-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathio_deisa.yml
176 lines (146 loc) · 5.51 KB
/
io_deisa.yml
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
full_datasets: &full_datasets
Time: { type: double }
d: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
E: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
dX: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
mx: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
my: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
mz: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
Bx: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
By: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
Bz: { type: array, subtype: double, size: ['$ncell[2]', '$ncell[1]', '$ncell[0]'] }
local_dataset_selection: &local_dataset_selection
size: ['$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: ['$start[2]', '$start[1]', '$start[0]']
d_description: &d_description
dataset: d
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [0, 0, 0, 0]
dataset_selection: *local_dataset_selection
E_description: &E_description
dataset: E
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [1, 0, 0, 0]
dataset_selection: *local_dataset_selection
mx_description: &mx_description
dataset: mx
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [2, 0, 0, 0]
dataset_selection: *local_dataset_selection
my_description: &my_description
dataset: my
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [3, 0, 0, 0]
dataset_selection: *local_dataset_selection
mz_description: &mz_description
dataset: mz
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [4, 0, 0, 0]
dataset_selection: *local_dataset_selection
Bx_description: &Bx_description
dataset: Bx
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [5, 0, 0, 0]
dataset_selection: *local_dataset_selection
By_description: &By_description
dataset: By
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [6, 0, 0, 0]
dataset_selection: *local_dataset_selection
Bz_description: &Bz_description
dataset: Bz
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [7, 0, 0, 0]
dataset_selection: *local_dataset_selection
dX_description: &dX_description
dataset: dX
memory_selection:
size: [1, '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [8, 0, 0, 0]
dataset_selection: *local_dataset_selection
pdi:
metadata:
nvar: int
ncell: { type: array, subtype: int, size: 3 }
ghost: { type: array, subtype: int, size: 3 }
ncell_local: { type: array, subtype: int, size: 3 }
start: { type: array, subtype: int, size: 3 }
origin: { type: array, subtype: double, size: 3 }
dl: { type: array, subtype: double, size: 3 }
gamma: double
mmw: double
prefix_size: int
prefix: { type: array, subtype: char, size: $prefix_size }
rank: int
size: int
time: double
iStep: int
mpi_coord: {type: array, subtype: int, size: 3}
data:
filename_size: int
filename: { type: array, subtype: char, size: $filename_size }
Rstar_h: double
output_id: int
restart_id: int
local_full_field:
type: array
subtype: double
size: ['${nvar}', '$ncell_local[2]+2*$ghost[2]', '$ncell_local[1]+2*$ghost[1]', '$ncell_local[0]+2*$ghost[0]']
subsize: ['${nvar}', '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [0, '$ghost[2]', '$ghost[1]', '$ghost[0]']
outputs_record_size: int
grid_size: { type: array, subtype: int, size: 3 }
plugins:
user_code:
on_event:
write_xml:
writeXML: {}
deisa:
scheduler_info: scheduler.json
init_on: init_pdi_w_deisa # Event called after sharing all metdata
time_step: $iStep
deisa_arrays: # Those are Deisa virtual arrays equivalent to Dask arrays
global_t: # That's the name that i will send to Dask for this array
type: array
subtype: double
size: [500, '${nvar}', '$ncell[2]', '$ncell[1]', '$ncell[0]']
subsize: [1, '${nvar}', '$ncell_local[2]', '$ncell_local[1]', '$ncell_local[0]']
start: [$iStep , 0, '$ncell_local[2]*$mpi_coord[2]', '$ncell_local[1]*$mpi_coord[1]', '$ncell_local[0]*$mpi_coord[0]']
+timedim: 0 # A tag for the time dim, only this configuration is supported for the moment
map_in: # Which local data will be mapped to which deisa array
local_full_field: global_t
mpi:
decl_hdf5:
- file: ${filename}
datasets: *full_datasets
on_event: checkpoint
collision_policy: write_into
when: '$iStep%100 = 0'
write:
grid_size:
time:
- dataset: Time
Rstar_h:
gamma:
mmw:
iStep:
output_id:
restart_id:
local_full_field:
- *d_description
- *E_description
- *mx_description
- *my_description
- *mz_description
- *Bx_description
- *By_description
- *Bz_description
- *dX_description