Add support for tasks per device setting for openfoam

pyepic/applications/base.py

@@ -20,7 +20,6 @@ class Distribution(Enum):
     NODE = "node"
     DEVICE = "device"
 
-
 class Upload(Enum):
     """Should excluded files be uploaded? Yes, No or only when the job finishes in an error state."""
 
@@ -56,6 +55,7 @@ def __init__(self, execute_step=True):
         self.execute = execute_step
         self.partitions = 1
         self.task_distribution = Distribution.CORE
+        self.tasks_per_device = 1
         self.runtime = 1
         self.run_if_previous_step_fails = True
         self.hyperthreading = True
@@ -72,6 +72,7 @@ def get_task_spec(self):
             runtime=self.runtime,
             task_distribution=self.task_distribution.value,
             hyperthreading=self.hyperthreading,
+            tasks_per_device=self.tasks_per_device
         )
         return spec
 

pyepic/applications/openfoam.py

@@ -62,6 +62,7 @@ def __init__(self, *args, **kwargs):
         self.endTime = 0
         self.startTime = 0
         self.application = "simpleFoam"
+        self.solver_tasks_per_device = 1
 
 
 class ReconstructParStep(JobStep):
@@ -147,6 +148,7 @@ def get_applications_options(self):
             "reconstruct_time": self.reconstructPar.reconstruct_time,
             "upload_excludes": self.sync_processor_directories.value,
             "solver_tasks_per_node": self.solver.task_distribution.value,
+            "solver_tasks_per_device": self.solver.tasks_per_device
         }
 
     def get_job_create_spec(self, queue_code):

setup.cfg

@@ -18,5 +18,5 @@ classifiers =
 packages = find:
 include_package_data = True
 install_requires =
-    epiccore>=0.0.25
+    epiccore>=0.0.27
     boto3>=1.16.57