Récupération de benchs pour H100; ajout de labels aux axes

idefix-code · Jan 24, 2025 · 49a56ee · 49a56ee
1 parent 76e399a
commit 49a56ee
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 2 deletions.
diff --git a/doc/source/bench-jeanzay.json b/doc/source/bench-jeanzay.json
@@ -142,5 +142,69 @@
         "cell_updates": 3.083581E+8
       }
     ]
+  },
+  {
+    "date": "2025-01-24_13:02:08",
+    "gpumodel": "h100",
+    "idefix_commit": "54482d582361f1cd05c39717902f752b3d04ae6e",
+    "bench_commit": "a94bc00490b3c62f81f7713327f4388e44a2abea",
+    "results": [
+      {
+        "nbgpu": 1,
+        "cell_updates": 3.076537E+8
+      }
+    ]
+  },
+  {
+    "date": "2025-01-24_13:04:02",
+    "gpumodel": "h100",
+    "idefix_commit": "54482d582361f1cd05c39717902f752b3d04ae6e",
+    "bench_commit": "a94bc00490b3c62f81f7713327f4388e44a2abea",
+    "results": [
+      {
+        "nbgpu": 1,
+        "cell_updates": 3.032686E+8
+      },
+      {
+        "nbgpu": 2,
+        "cell_updates": 3.007089E+8
+      },
+      {
+        "nbgpu": 4,
+        "cell_updates": 2.938962E+8
+      }
+    ]
+  },
+  {
+    "date": "2025-01-24_14:17:30",
+    "gpumodel": "h100",
+    "idefix_commit": "54482d582361f1cd05c39717902f752b3d04ae6e",
+    "bench_commit": "a94bc00490b3c62f81f7713327f4388e44a2abea",
+    "results": [
+      {
+        "nbgpu": 1,
+        "cell_updates": 3.074363E+8
+      },
+      {
+        "nbgpu": 2,
+        "cell_updates": 3.009340E+8
+      },
+      {
+        "nbgpu": 4,
+        "cell_updates": 2.938868E+8
+      },
+      {
+        "nbgpu": 8,
+        "cell_updates": 2.833054E+8
+      },
+      {
+        "nbgpu": 16,
+        "cell_updates": 2.822483E+8
+      },
+      {
+        "nbgpu": 32,
+        "cell_updates": 2.817151E+8
+      }
+    ]
   }
 ]
diff --git a/doc/source/plot_idefix_bench.py b/doc/source/plot_idefix_bench.py
@@ -8,7 +8,10 @@ def do_plot(title, bench_file, gpumodel):
     select = [bench for bench in benches if bench['gpumodel'] == gpumodel][-1]
     xs = [r['nbgpu'] for r in select['results']]
     ys = [r['cell_updates'] for r in select['results']]
-    
-    plt.xscale("log")
+
+    plt.xscale("log", base=2)
     plt.plot(xs, ys)
+    plt.ylim(0,max(ys)*1.1)
+    plt.xlabel("Number of GPUs")
+    plt.ylabel("Performance (cells / second / GPU)")
     plt.title(title)