From 239485967171490d232a727d48e6fc4059f703a4 Mon Sep 17 00:00:00 2001
From: Lin Guo <linsword13@gmail.com>
Date: Tue, 15 Oct 2024 11:39:03 -0700
Subject: [PATCH] Add Exchange workload to imb

---
 .../intel-mpi-benchmarks/application.py       | 25 ++++++++++++++-----
 1 file changed, 19 insertions(+), 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
index c770f2543..1d350181c 100644
--- 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: