+ strong scaling

preprocess/generator.py

@@ -13,9 +13,9 @@ if __name__ == "__main__":
     args = parser().parse_args()
     with open(args.output, "w") as f:
         if args.machine == "jurecadc":
-            f.write("24")
-        elif args.machine == "jedi": 
             f.write("12")
+        elif args.machine == "jedi": 
+            f.write("24")
         else:
             raise ValueError("Unknown machine")
         

simple.yml

@@ -42,13 +42,13 @@ parameterset:
     parameter:
       - name: nodes
         mode: python
-        _: '{ "jurecadc" : 2, "jedi" : 1 }["${system_name}"]'
+        _: '{ "jurecadc" : "2,4" , "jedi" : "1,2" }["${system_name}"]'
       - name: queue
         mode: python
         _: '{ "jurecadc" : "dc-cpu", "jedi" : "all" }["${system_name}"]'
       - name: preprocess
         mode: python
-        _: '{ "jurecadc" : "module load Stages/2023 && module load GCC Python/3.10.4", "jedi" : "module load GCC Python/3.11.3" }["${system_name}"]'
+        _: '{ "jurecadc" : "module load GCC Python OpenMPI mpi4py", "jedi" : "module load GCC Python OpenMPI mpi4py" }["${system_name}"]'
       - name: executable
         _: "python"
       - name: args_exec

src/value.py

@@ -2,6 +2,7 @@
 
 import argparse
 import random
+from mpi4py import MPI
 
 def parser():
     parser = argparse.ArgumentParser(description="Value of a number")
@@ -10,9 +11,13 @@ def parser():
     return parser
 
 if __name__ == "__main__":
+    comm = MPI.COMM_WORLD
+    nproc = comm.Get_size()
     args = parser().parse_args()
     with open(args.input, "r") as f:
         data = f.read()
-        runtime = float(data) * float(args.number) * 12
+        problem_size = float(data) * float(args.number) * 12
+        runtime = problem_size / nproc * random.uniform(0.8, 1.2)
         string = " " * random.randint(1, 10)
+        if comm.Get_rank() == 0:
             print("Runtime={}{}".format(string,runtime))