Skip to content

Commit 112cd22

Browse files
sjpbsjpb
committed
add roce/ib ratio to WRF
1 parent 1c5ab9b commit 112cd22

File tree

1 file changed

+113
-38
lines changed

1 file changed

+113
-38
lines changed

‎apps/wrf/wrf.ipynb

+113-38
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
},
1212
{
1313
"cell_type": "code",
14-
"execution_count": 3,
14+
"execution_count": 1,
1515
"metadata": {},
1616
"outputs": [],
1717
"source": [
@@ -41,7 +41,7 @@
4141
},
4242
{
4343
"cell_type": "code",
44-
"execution_count": 4,
44+
"execution_count": 2,
4545
"metadata": {},
4646
"outputs": [
4747
{
@@ -76,63 +76,82 @@
7676
" <th>case</th>\n",
7777
" <th>csd3:cclake-ib-icc19-impi19-ucx</th>\n",
7878
" <th>csd3:cclake-roce-icc19-impi19-ucx</th>\n",
79+
" <th>csd3:cclake-[roce/ib]-icc19-impi19-ucx</th>\n",
7980
" </tr>\n",
8081
" <tr>\n",
8182
" <th>num_nodes</th>\n",
8283
" <th></th>\n",
8384
" <th></th>\n",
85+
" <th></th>\n",
8486
" </tr>\n",
8587
" </thead>\n",
8688
" <tbody>\n",
8789
" <tr>\n",
8890
" <th>1</th>\n",
8991
" <td>58.267</td>\n",
9092
" <td>58.541</td>\n",
93+
" <td>1.00</td>\n",
9194
" </tr>\n",
9295
" <tr>\n",
9396
" <th>2</th>\n",
9497
" <td>36.250</td>\n",
9598
" <td>35.471</td>\n",
99+
" <td>0.98</td>\n",
96100
" </tr>\n",
97101
" <tr>\n",
98102
" <th>4</th>\n",
99103
" <td>31.304</td>\n",
100104
" <td>30.315</td>\n",
105+
" <td>0.97</td>\n",
101106
" </tr>\n",
102107
" <tr>\n",
103108
" <th>8</th>\n",
104109
" <td>18.424</td>\n",
105110
" <td>82.590</td>\n",
111+
" <td>4.48</td>\n",
106112
" </tr>\n",
107113
" <tr>\n",
108114
" <th>16</th>\n",
109115
" <td>17.148</td>\n",
110116
" <td>82.629</td>\n",
117+
" <td>4.82</td>\n",
111118
" </tr>\n",
112119
" <tr>\n",
113120
" <th>32</th>\n",
114121
" <td>19.816</td>\n",
115122
" <td>18.087</td>\n",
123+
" <td>0.91</td>\n",
116124
" </tr>\n",
117125
" <tr>\n",
118126
" <th>56</th>\n",
119127
" <td>22.920</td>\n",
120128
" <td>21.972</td>\n",
129+
" <td>0.96</td>\n",
121130
" </tr>\n",
122131
" </tbody>\n",
123132
"</table>\n",
124133
"</div>"
125134
],
126135
"text/plain": [
127-
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx\n",
128-
"num_nodes \n",
129-
"1 58.267 58.541\n",
130-
"2 36.250 35.471\n",
131-
"4 31.304 30.315\n",
132-
"8 18.424 82.590\n",
133-
"16 17.148 82.629\n",
134-
"32 19.816 18.087\n",
135-
"56 22.920 21.972"
136+
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx \\\n",
137+
"num_nodes \n",
138+
"1 58.267 58.541 \n",
139+
"2 36.250 35.471 \n",
140+
"4 31.304 30.315 \n",
141+
"8 18.424 82.590 \n",
142+
"16 17.148 82.629 \n",
143+
"32 19.816 18.087 \n",
144+
"56 22.920 21.972 \n",
145+
"\n",
146+
"case csd3:cclake-[roce/ib]-icc19-impi19-ucx \n",
147+
"num_nodes \n",
148+
"1 1.00 \n",
149+
"2 0.98 \n",
150+
"4 0.97 \n",
151+
"8 4.48 \n",
152+
"16 4.82 \n",
153+
"32 0.91 \n",
154+
"56 0.96 "
136155
]
137156
},
138157
"metadata": {},
@@ -170,63 +189,82 @@
170189
" <th>case</th>\n",
171190
" <th>csd3:cclake-ib-icc19-impi19-ucx</th>\n",
172191
" <th>csd3:cclake-roce-icc19-impi19-ucx</th>\n",
192+
" <th>csd3:cclake-[roce/ib]-icc19-impi19-ucx</th>\n",
173193
" </tr>\n",
174194
" <tr>\n",
175195
" <th>num_nodes</th>\n",
176196
" <th></th>\n",
177197
" <th></th>\n",
198+
" <th></th>\n",
178199
" </tr>\n",
179200
" </thead>\n",
180201
" <tbody>\n",
181202
" <tr>\n",
182203
" <th>1</th>\n",
183204
" <td>100.738468</td>\n",
184205
" <td>100.780792</td>\n",
206+
" <td>1.00</td>\n",
185207
" </tr>\n",
186208
" <tr>\n",
187209
" <th>2</th>\n",
188210
" <td>209.806785</td>\n",
189211
" <td>209.858477</td>\n",
212+
" <td>1.00</td>\n",
190213
" </tr>\n",
191214
" <tr>\n",
192215
" <th>4</th>\n",
193216
" <td>390.281351</td>\n",
194217
" <td>388.785974</td>\n",
218+
" <td>1.00</td>\n",
195219
" </tr>\n",
196220
" <tr>\n",
197221
" <th>8</th>\n",
198222
" <td>759.254805</td>\n",
199223
" <td>729.208732</td>\n",
224+
" <td>0.96</td>\n",
200225
" </tr>\n",
201226
" <tr>\n",
202227
" <th>16</th>\n",
203228
" <td>1063.881433</td>\n",
204229
" <td>929.652122</td>\n",
230+
" <td>0.87</td>\n",
205231
" </tr>\n",
206232
" <tr>\n",
207233
" <th>32</th>\n",
208234
" <td>1419.716396</td>\n",
209235
" <td>1151.612089</td>\n",
236+
" <td>0.81</td>\n",
210237
" </tr>\n",
211238
" <tr>\n",
212239
" <th>56</th>\n",
213240
" <td>1184.661126</td>\n",
214241
" <td>1151.982362</td>\n",
242+
" <td>0.97</td>\n",
215243
" </tr>\n",
216244
" </tbody>\n",
217245
"</table>\n",
218246
"</div>"
219247
],
220248
"text/plain": [
221-
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx\n",
222-
"num_nodes \n",
223-
"1 100.738468 100.780792\n",
224-
"2 209.806785 209.858477\n",
225-
"4 390.281351 388.785974\n",
226-
"8 759.254805 729.208732\n",
227-
"16 1063.881433 929.652122\n",
228-
"32 1419.716396 1151.612089\n",
229-
"56 1184.661126 1151.982362"
249+
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx \\\n",
250+
"num_nodes \n",
251+
"1 100.738468 100.780792 \n",
252+
"2 209.806785 209.858477 \n",
253+
"4 390.281351 388.785974 \n",
254+
"8 759.254805 729.208732 \n",
255+
"16 1063.881433 929.652122 \n",
256+
"32 1419.716396 1151.612089 \n",
257+
"56 1184.661126 1151.982362 \n",
258+
"\n",
259+
"case csd3:cclake-[roce/ib]-icc19-impi19-ucx \n",
260+
"num_nodes \n",
261+
"1 1.00 \n",
262+
"2 1.00 \n",
263+
"4 1.00 \n",
264+
"8 0.96 \n",
265+
"16 0.87 \n",
266+
"32 0.81 \n",
267+
"56 0.97 "
230268
]
231269
},
232270
"metadata": {},
@@ -264,57 +302,74 @@
264302
" <th>case</th>\n",
265303
" <th>csd3:cclake-ib-icc19-impi19-ucx</th>\n",
266304
" <th>csd3:cclake-roce-icc19-impi19-ucx</th>\n",
305+
" <th>csd3:cclake-[roce/ib]-icc19-impi19-ucx</th>\n",
267306
" </tr>\n",
268307
" <tr>\n",
269308
" <th>num_nodes</th>\n",
270309
" <th></th>\n",
271310
" <th></th>\n",
311+
" <th></th>\n",
272312
" </tr>\n",
273313
" </thead>\n",
274314
" <tbody>\n",
275315
" <tr>\n",
276316
" <th>1</th>\n",
277317
" <td>2862.343</td>\n",
278318
" <td>2851.405</td>\n",
319+
" <td>1.00</td>\n",
279320
" </tr>\n",
280321
" <tr>\n",
281322
" <th>2</th>\n",
282323
" <td>1504.685</td>\n",
283324
" <td>1506.094</td>\n",
325+
" <td>1.00</td>\n",
284326
" </tr>\n",
285327
" <tr>\n",
286328
" <th>8</th>\n",
287329
" <td>477.870</td>\n",
288330
" <td>706.603</td>\n",
331+
" <td>1.48</td>\n",
289332
" </tr>\n",
290333
" <tr>\n",
291334
" <th>16</th>\n",
292335
" <td>313.148</td>\n",
293336
" <td>802.775</td>\n",
337+
" <td>2.56</td>\n",
294338
" </tr>\n",
295339
" <tr>\n",
296340
" <th>32</th>\n",
297341
" <td>229.541</td>\n",
298342
" <td>1233.355</td>\n",
343+
" <td>5.37</td>\n",
299344
" </tr>\n",
300345
" <tr>\n",
301346
" <th>56</th>\n",
302347
" <td>198.619</td>\n",
303348
" <td>879.385</td>\n",
349+
" <td>4.43</td>\n",
304350
" </tr>\n",
305351
" </tbody>\n",
306352
"</table>\n",
307353
"</div>"
308354
],
309355
"text/plain": [
310-
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx\n",
311-
"num_nodes \n",
312-
"1 2862.343 2851.405\n",
313-
"2 1504.685 1506.094\n",
314-
"8 477.870 706.603\n",
315-
"16 313.148 802.775\n",
316-
"32 229.541 1233.355\n",
317-
"56 198.619 879.385"
356+
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx \\\n",
357+
"num_nodes \n",
358+
"1 2862.343 2851.405 \n",
359+
"2 1504.685 1506.094 \n",
360+
"8 477.870 706.603 \n",
361+
"16 313.148 802.775 \n",
362+
"32 229.541 1233.355 \n",
363+
"56 198.619 879.385 \n",
364+
"\n",
365+
"case csd3:cclake-[roce/ib]-icc19-impi19-ucx \n",
366+
"num_nodes \n",
367+
"1 1.00 \n",
368+
"2 1.00 \n",
369+
"8 1.48 \n",
370+
"16 2.56 \n",
371+
"32 5.37 \n",
372+
"56 4.43 "
318373
]
319374
},
320375
"metadata": {},
@@ -352,57 +407,74 @@
352407
" <th>case</th>\n",
353408
" <th>csd3:cclake-ib-icc19-impi19-ucx</th>\n",
354409
" <th>csd3:cclake-roce-icc19-impi19-ucx</th>\n",
410+
" <th>csd3:cclake-[roce/ib]-icc19-impi19-ucx</th>\n",
355411
" </tr>\n",
356412
" <tr>\n",
357413
" <th>num_nodes</th>\n",
358414
" <th></th>\n",
359415
" <th></th>\n",
416+
" <th></th>\n",
360417
" </tr>\n",
361418
" </thead>\n",
362419
" <tbody>\n",
363420
" <tr>\n",
364421
" <th>1</th>\n",
365422
" <td>108.277093</td>\n",
366423
" <td>108.606037</td>\n",
424+
" <td>1.00</td>\n",
367425
" </tr>\n",
368426
" <tr>\n",
369427
" <th>2</th>\n",
370428
" <td>214.725493</td>\n",
371429
" <td>214.697360</td>\n",
430+
" <td>1.00</td>\n",
372431
" </tr>\n",
373432
" <tr>\n",
374433
" <th>8</th>\n",
375434
" <td>832.436320</td>\n",
376435
" <td>769.011059</td>\n",
436+
" <td>0.92</td>\n",
377437
" </tr>\n",
378438
" <tr>\n",
379439
" <th>16</th>\n",
380440
" <td>1561.541929</td>\n",
381441
" <td>457.541821</td>\n",
442+
" <td>0.29</td>\n",
382443
" </tr>\n",
383444
" <tr>\n",
384445
" <th>32</th>\n",
385446
" <td>3170.396797</td>\n",
386447
" <td>271.847855</td>\n",
448+
" <td>0.09</td>\n",
387449
" </tr>\n",
388450
" <tr>\n",
389451
" <th>56</th>\n",
390452
" <td>4992.934844</td>\n",
391453
" <td>405.367378</td>\n",
454+
" <td>0.08</td>\n",
392455
" </tr>\n",
393456
" </tbody>\n",
394457
"</table>\n",
395458
"</div>"
396459
],
397460
"text/plain": [
398-
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx\n",
399-
"num_nodes \n",
400-
"1 108.277093 108.606037\n",
401-
"2 214.725493 214.697360\n",
402-
"8 832.436320 769.011059\n",
403-
"16 1561.541929 457.541821\n",
404-
"32 3170.396797 271.847855\n",
405-
"56 4992.934844 405.367378"
461+
"case csd3:cclake-ib-icc19-impi19-ucx csd3:cclake-roce-icc19-impi19-ucx \\\n",
462+
"num_nodes \n",
463+
"1 108.277093 108.606037 \n",
464+
"2 214.725493 214.697360 \n",
465+
"8 832.436320 769.011059 \n",
466+
"16 1561.541929 457.541821 \n",
467+
"32 3170.396797 271.847855 \n",
468+
"56 4992.934844 405.367378 \n",
469+
"\n",
470+
"case csd3:cclake-[roce/ib]-icc19-impi19-ucx \n",
471+
"num_nodes \n",
472+
"1 1.00 \n",
473+
"2 1.00 \n",
474+
"8 0.92 \n",
475+
"16 0.29 \n",
476+
"32 0.09 \n",
477+
"56 0.08 "
406478
]
407479
},
408480
"metadata": {},
@@ -466,12 +538,15 @@
466538
" continue\n",
467539
" title = '%s: %s' % (test, var)\n",
468540
" display_markdown('### %s' % title, raw=True)\n",
469-
" display(df)\n",
470541
" ax = df.plot(grid=True, style='o-')\n",
471542
" ax.set_xlabel('number of nodes')\n",
472543
" ax.set_ylabel(var)\n",
473544
" ax.set_title(title)\n",
474-
" plt.tight_layout()"
545+
" plt.tight_layout()\n",
546+
" \n",
547+
" modules.plots.add_roce_ib_factor(df)\n",
548+
" display(df)\n",
549+
" "
475550
]
476551
},
477552
{

0 commit comments

Comments
 (0)