Merge pull request #232 from DavidHuber-NOAA/port_master_2_jet

GitHub Issue NOAA-EMC/GSI#215. Update Jet support, port monitors to Jet and S4.
NOAA-EMC · Oct 15, 2021 · ebd9dd7 · ebd9dd7
2 parents b8a6818 + 260f6d1
commit ebd9dd7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ush/calcanl_gfs.py b/ush/calcanl_gfs.py
@@ -229,6 +229,13 @@ def calcanl_gfs(DoIAU, l4DEnsVar, Write4Danl, ComOut, APrefix, ASuffix,
             with open(CalcAnlDir+'/hosts', 'w') as hostfile:
                 hostfile.write(hosts[ihost]+'\n')
                 if launcher == 'srun': # need to write host per task not per node for slurm
+                    # For xjet, each instance of chgres_inc must run on two nodes each
+                    if os.getenv('SLURM_JOB_PARTITION','') == 'xjet':
+                        for a in range(0,4):
+                            hostfile.write(hosts[ihost]+'\n')
+                        ihost+=1
+                        for a in range(0,5):
+                            hostfile.write(hosts[ihost]+'\n')
                     for a in range(0,9): # need 9 more of the same host for the 10 tasks for chgres_inc
                         hostfile.write(hosts[ihost]+'\n')
             if launcher == 'srun':