Faster Phoenix CPU benchmarking (#729)

MFlowCode · Nov 13, 2024 · 6578923 · 6578923
1 parent b41386d
commit 6578923
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 11 deletions.
diff --git a/.github/workflows/phoenix/bench.sh b/.github/workflows/phoenix/bench.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-n_ranks=4
+n_ranks=12
 
 if [ "$job_device" == "gpu" ]; then
     n_ranks=$(nvidia-smi -L | wc -l)        # number of GPUs on node
@@ -9,7 +9,7 @@ if [ "$job_device" == "gpu" ]; then
 fi
 
 if ["$job_device" == "gpu"]; then
-    ./mfc.sh bench --mem 8 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
+    ./mfc.sh bench --mem 12 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
 else
     ./mfc.sh bench --mem 1 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
-fi
+fi
diff --git a/benchmarks/5eq_rk3_weno3_hllc/case.py b/benchmarks/5eq_rk3_weno3_hllc/case.py
@@ -194,8 +194,8 @@
     'cyl_coord'                    : 'F',
     'dt'                           : dt,
     't_step_start'                 : 0,
-    't_step_stop'                  : int(60*(95*size + 5)),
-    't_step_save'                  : int(60*(95*size + 5)),
+    't_step_stop'                  : int(30*(95*size + 5)),
+    't_step_save'                  : int(30*(95*size + 5)),
     # ==========================================================
 
     # Simulation Algorithm Parameters ==========================

diff --git a/benchmarks/hypo_hll/case.py b/benchmarks/hypo_hll/case.py
@@ -47,8 +47,8 @@
                     'p'                            : Nz,
                     'dt'                           : 1e-8,
                     't_step_start'                 : 0,
-                    't_step_stop'                  : int(60*(95*size + 5)),
-                    't_step_save'                  : int(60*(95*size + 5)),
+                    't_step_stop'                  : int(30*(95*size + 5)),
+                    't_step_save'                  : int(30*(95*size + 5)),
 		    # ==========================================================
 
                     # Simulation Algorithm Parameters ==========================

diff --git a/benchmarks/ibm/case.py b/benchmarks/ibm/case.py
@@ -51,8 +51,8 @@
             'p'                            : Nz,                         
             'dt'                           : mydt,                      
             't_step_start'                 : 0,                         
-            't_step_stop'                  : int(40*(95*size + 5)),
-            't_step_save'                  : int(40*(95*size + 5)),
+            't_step_stop'                  : int(20*(95*size + 5)),
+            't_step_save'                  : int(20*(95*size + 5)),
 		    # ==========================================================
 
             # Simulation Algorithm Parameters ==========================

diff --git a/benchmarks/viscous_weno5_sgb_acoustic/case.py b/benchmarks/viscous_weno5_sgb_acoustic/case.py
@@ -113,8 +113,8 @@
     'p'                            : Nz,
     'dt'                           : dt,
     't_step_start'                 : 0,
-    't_step_stop'                  : int(30*(25*size + 5)),
-    't_step_save'                  : int(30*(25*size + 5)),
+    't_step_stop'                  : int(15*(25*size + 5)),
+    't_step_save'                  : int(15*(25*size + 5)),
     # ==========================================================
 
     # Simulation Algorithm Parameters ==========================