account for threads in override case

ESMCI · Oct 10, 2024 · f0b5c09 · f0b5c09
1 parent 5271a4e
commit f0b5c09
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/CIME/XML/env_batch.py b/CIME/XML/env_batch.py
@@ -223,6 +223,9 @@ def get_job_overrides(self, job, case):
             overrides["tasks_per_node"] = tasks_per_node
             if thread_count:
                 overrides["thread_count"] = thread_count
+                total_tasks = total_tasks * thread_count
+            else:
+                total_tasks = total_tasks * case.thread_count
         else:
             # Total PES accounts for threads as well as mpi tasks
             total_tasks = case.get_value("TOTALPES")