feat(tasks): allow Compute4PUNCH backend options (#210)

reanahub · Jul 4, 2024 · a6313f2 · a6313f2
1 parent 009a57a
commit a6313f2
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/AUTHORS.md b/AUTHORS.md
@@ -11,6 +11,7 @@ The list of contributors in alphabetical order:
 - [Dinos Kousidis](https://orcid.org/0000-0002-4914-4289)
 - [Giuseppe Steduto](https://orcid.org/0009-0002-1258-8553)
 - [Jan Okraska](https://orcid.org/0000-0002-1416-3244)
+- [Manuel Giffels](https://orcid.org/0000-0003-0193-3032)
 - [Marco Donadoni](https://orcid.org/0000-0003-2922-5505)
 - [Marco Vidal](https://orcid.org/0000-0002-9363-4971)
 - [Rokas Maciulaitis](https://orcid.org/0000-0003-1064-6967)

diff --git a/reana_workflow_engine_serial/tasks.py b/reana_workflow_engine_serial/tasks.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 #
 # This file is part of REANA.
-# Copyright (C) 2018, 2019, 2020, 2021, 2022 CERN.
+# Copyright (C) 2018, 2019, 2020, 2021, 2022, 2023, 2024 CERN.
 #
 # REANA is free software; you can redistribute it and/or modify it
 # under the terms of the MIT License; see LICENSE file for more details.
@@ -124,6 +124,9 @@ def run_step(
             htcondor_accounting_group=step.get("htcondor_accounting_group", ""),
             slurm_partition=step.get("slurm_partition"),
             slurm_time=step.get("slurm_time"),
+            c4p_cpu_cores=step.get("c4p_cpu_cores"),
+            c4p_memory_limit=step.get("c4p_memory_limit"),
+            c4p_additional_requirements=step.get("c4p_additional_requirements"),
         )
         job_spec_copy = dict(job_spec)
         job_spec_copy["cmd"] = command

diff --git a/reana_workflow_engine_serial/utils.py b/reana_workflow_engine_serial/utils.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 #
 # This file is part of REANA.
-# Copyright (C) 2019, 2020, 2021, 2022 CERN.
+# Copyright (C) 2019, 2020, 2021, 2022, 2023, 2024 CERN.
 #
 # REANA is free software; you can redistribute it and/or modify it
 # under the terms of the MIT License; see LICENSE file for more details.
@@ -36,6 +36,9 @@ def build_job_spec(
     htcondor_accounting_group,
     slurm_partition,
     slurm_time,
+    c4p_cpu_cores,
+    c4p_memory_limit,
+    c4p_additional_requirements,
 ):
     """Build job specification to passed to RJC."""
     job_spec = {
@@ -58,6 +61,9 @@ def build_job_spec(
         "htcondor_accounting_group": htcondor_accounting_group,
         "slurm_partition": slurm_partition,
         "slurm_time": slurm_time,
+        "c4p_cpu_cores": c4p_cpu_cores,
+        "c4p_memory_limit": c4p_memory_limit,
+        "c4p_additional_requirements": c4p_additional_requirements,
     }
     return job_spec