Skip to content

Commit be02afa

Browse files
fix Torque header format (#205)
* fix Torque header format * add ref
1 parent 31091de commit be02afa

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

dpdispatcher/pbs.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,3 +114,16 @@ def check_status(self, job):
114114
return JobStatus.terminated
115115
else :
116116
return JobStatus.unknown
117+
118+
def gen_script_header(self, job):
119+
# ref: https://support.adaptivecomputing.com/wp-content/uploads/2021/02/torque/torque.htm#topics/torque/2-jobs/requestingRes.htm
120+
resources = job.resources
121+
pbs_script_header_dict= {}
122+
pbs_script_header_dict['select_node_line']="#PBS -l nodes={number_node}:ppn={cpu_per_node}".format(
123+
number_node=resources.number_node, cpu_per_node=resources.cpu_per_node
124+
)
125+
if (resources.gpu_per_node != 0):
126+
pbs_script_header_dict['select_node_line'] += ":gpus={gpu_per_node}".format(gpu_per_node=resources.gpu_per_node)
127+
pbs_script_header_dict['queue_name_line']="#PBS -q {queue_name}".format(queue_name=resources.queue_name)
128+
pbs_script_header = pbs_script_header_template.format(**pbs_script_header_dict)
129+
return pbs_script_header

0 commit comments

Comments
 (0)