Merge pull request #228 from wguanicedew/master

fix number of cores
HSF · May 16, 2024 · 0142dd5 · 0142dd5
2 parents a47e2c1 + 49d9840
commit 0142dd5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pandaharvester/harvestersubmitter/slurm_submitter.py b/pandaharvester/harvestersubmitter/slurm_submitter.py
@@ -40,7 +40,7 @@ def submit_workers(self, workspec_list):
             # make logger
             tmpLog = self.make_logger(baseLogger, f"workerID={workSpec.workerID}", method_name="submit_workers")
             # set nCore
-            workSpec.nCore = self.nCore
+            # workSpec.nCore = self.nCore
             # make batch script
             batchFile = self.make_batch_script(workSpec)
             # command
@@ -90,6 +90,8 @@ def make_placeholder_map(self, workspec):
             n_core_per_node = self.nCorePerNode if self.nCorePerNode else n_core_per_node_from_queue
         except AttributeError:
             n_core_per_node = n_core_per_node_from_queue
+        if not n_core_per_node:
+            n_core_per_node = self.nCore
 
         n_core_total = workspec.nCore if workspec.nCore else n_core_per_node
         n_core_total_factor = n_core_total * self.nCoreFactor