hook in the new resolution

AMReX-Astro · Jul 9, 2024 · ca27968 · ca27968
1 parent 62995f7
commit ca27968
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 30 deletions.
diff --git a/Exec/science/flame_wave/scaling/frontier/frontier-scaling-2024-07-04-subch_simple.txt b/Exec/science/flame_wave/scaling/frontier/frontier-scaling-2024-07-04-subch_simple.txt
@@ -27,7 +27,7 @@
   64      6.0          128           167.502   0.419448
  128      6.0          128            85.4082  0.29416
  256      6.0          128            46.4092  0.876868
- 384      6.0          128            37.5447  0.138806
+# 384      6.0          128            37.5447  0.138806
  512      6.0          128            25.5446  0.123848
 1024      6.0          128            17.3517  0.0857189
 #2048      6.0          128

diff --git a/Exec/science/flame_wave/scaling/frontier/frontier-scaling-2024-07-04.txt b/Exec/science/flame_wave/scaling/frontier/frontier-scaling-2024-07-04.txt
@@ -26,7 +26,7 @@
   64      6.0          128           50.2029   0.681104
  128      6.0          128           28.4001   0.221368
  256      6.0          128           15.7771   0.133056
- 384      6.0          128           12.8213   0.37406
+# 384      6.0          128           12.8213   0.37406
  512      6.0          128            9.6077   0.331669
 1024      6.0          128            6.66329  0.103599
 #2048      6.0          128

diff --git a/Exec/science/flame_wave/scaling/frontier/frontier-scaling-rkc-2024-07-04.txt b/Exec/science/flame_wave/scaling/frontier/frontier-scaling-rkc-2024-07-04.txt
@@ -26,7 +26,7 @@
   64      6.0          128           39.4336   0.32115
  128      6.0          128           22.8627   0.22725
  256      6.0          128           12.9558   0.12379
- 384      6.0          128           10.3259   0.123708
+# 384      6.0          128           10.3259   0.123708
  512      6.0          128            7.9563   0.104311
 1024      6.0          128            5.65801  0.109306
 #2048      6.0          128

diff --git a/Exec/science/flame_wave/scaling/frontier/frontier_flame_wave_scaling.png b/Exec/science/flame_wave/scaling/frontier/frontier_flame_wave_scaling.png
diff --git a/Exec/science/flame_wave/scaling/frontier/frontier_flame_wave_scaling_by_gpus.png b/Exec/science/flame_wave/scaling/frontier/frontier_flame_wave_scaling_by_gpus.png
diff --git a/Exec/science/flame_wave/scaling/frontier/frontier_scaling.py b/Exec/science/flame_wave/scaling/frontier/frontier_scaling.py
@@ -27,6 +27,12 @@
 summit_times = summit_data[:, 2]
 summit_std = summit_data[:, 3]
 
+frontier_bignet_data = np.loadtxt("frontier-scaling-2024-07-04-subch_simple.txt")
+
+frontier_bignet_nodes = frontier_bignet_data[:, 0]
+frontier_bignet_times = frontier_bignet_data[:, 3]
+frontier_bignet_std = frontier_bignet_data[:, 4]
+
 
 def trend_line(c, t):
     cnew = np.array(sorted(list(set(c))))
@@ -39,43 +45,30 @@ def trend_line(c, t):
 
 fig, ax = plt.subplots(1)
 
-ax.errorbar(frontier_nodes, frontier_times, yerr=frontier_std, ls="None", marker="x", label="Frontier (ROCm 5.3)")
-ax.errorbar(frontier_rkc_nodes, frontier_rkc_times, yerr=frontier_rkc_std, ls="None", marker="x", label="Frontier (RKC integrator)")
-ax.errorbar(summit_nodes, summit_times, yerr=summit_std, ls="None", marker="^", label="Summit (CUDA 11.4)")
+ax.errorbar(frontier_nodes, frontier_times, yerr=frontier_std,
+            ls="None", marker="x", label="Frontier (ROCm 6.0)")
+ax.errorbar(frontier_rkc_nodes, frontier_rkc_times, yerr=frontier_rkc_std,
+            ls="None", marker="x", label="Frontier (ROCm 6.0; RKC integrator)")
+ax.errorbar(summit_nodes, summit_times, yerr=summit_std,
+            ls="None", marker="^", label="Summit (CUDA 11.4)")
+ax.errorbar(frontier_bignet_nodes, frontier_bignet_times, yerr=frontier_bignet_std,
+            ls="None", marker="o", label="Frontier (ROCm 6.0; big network)")
 
 c, t = trend_line(frontier_nodes, frontier_times)
 ax.plot(c, t, alpha=0.5, linestyle=":")
 
-ax.set_ylabel("wallclock time / step")
-ax.set_xlabel("number of nodes")
-
-ax.set_xscale("log")
-ax.set_yscale("log")
-
-ax.legend()
-
-fig.savefig("frontier_flame_wave_scaling.png")
-
-
-# now by GPUs
-
-fig, ax = plt.subplots(1)
-
-nfrontier_gpu = 8
-nsummit_gpu = 6
-
-ax.errorbar(frontier_nodes * nfrontier_gpu, frontier_times, yerr=frontier_std, ls="None", marker="x", label="Frontier (ROCm 5.3)")
-ax.errorbar(summit_nodes * nsummit_gpu, summit_times, yerr=summit_std, ls="None", marker="x", label="Summit (CUDA 11.4)")
-
-c, t = trend_line(frontier_nodes * nfrontier_gpu, frontier_times)
+c, t = trend_line(frontier_bignet_nodes, frontier_bignet_times)
 ax.plot(c, t, alpha=0.5, linestyle=":")
 
+
 ax.set_ylabel("wallclock time / step")
-ax.set_xlabel("number of GPUs")
+ax.set_xlabel("number of nodes")
 
 ax.set_xscale("log")
 ax.set_yscale("log")
 
 ax.legend()
 
-fig.savefig("frontier_flame_wave_scaling_by_gpus.png")
+ax.set_title("3D XRB flame scaling")
+
+fig.savefig("frontier_flame_wave_scaling.png")