Merge pull request #697 from linsword13/imb

Add Exchange workload to imb
GoogleCloudPlatform · Oct 15, 2024 · bf8478b · bf8478b
2 parents 3f9a04f + 2394859
commit bf8478b
Showing 1 changed file with 19 additions and 6 deletions.
diff --git a/var/ramble/repos/builtin/applications/intel-mpi-benchmarks/application.py b/var/ramble/repos/builtin/applications/intel-mpi-benchmarks/application.py
@@ -56,9 +56,22 @@ class IntelMpiBenchmarks(ExecutableApplication):
         use_mpi=True,
     )
 
+    executable(
+        "exchange",
+        "{install_path}/IMB-MPI1 Exchange -msglog {msglog_min}:{msglog_max} "
+        "-iter {num_iterations} {additional_args}",
+        use_mpi=True,
+    )
+
     workload("pingpong", executable="pingpong")
     workload("multi-pingpong", executable="multi-pingpong")
     workload("collective", executable="collective")
+    workload("exchange", executable="exchange")
+
+    workload_group(
+        "mpi1",
+        workloads=["pingpong", "multi-pingpong", "collective", "exchange"],
+    )
 
     workload_variable(
         "num_cores",
@@ -97,7 +110,7 @@ class IntelMpiBenchmarks(ExecutableApplication):
         "install_path",
         default="{intel-mpi-benchmarks_path}/bin",
         description="User configurable dir to executables",
-        workloads=["pingpong", "multi-pingpong", "collective"],
+        workloads=["*"],
     )
 
     workload_variable(
@@ -149,28 +162,28 @@ class IntelMpiBenchmarks(ExecutableApplication):
         "num_iterations",
         default="1000",
         description="Number of iterations to test over",
-        workloads=["pingpong", "multi-pingpong", "collective"],
+        workload_group="mpi1",
     )
 
     workload_variable(
         "msglog_min",
         default="1",
         description="Min Message Size (power of 2)",
-        workloads=["pingpong", "multi-pingpong", "collective"],
+        workload_group="mpi1",
     )
 
     workload_variable(
         "msglog_max",
         default="30",
         description="Max Message Size (power of 2)",
-        workloads=["pingpong", "multi-pingpong", "collective"],
+        workload_group="mpi1",
     )
 
     workload_variable(
         "additional_args",
         default="",
-        description="Number of iterations to test over",
-        workloads=["pingpong", "multi-pingpong", "collective"],
+        description="Additional cmd-line arguments",
+        workload_group="mpi1",
     )
 
     # Matches tables like: