File tree Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Expand file tree Collapse file tree 1 file changed +13
-0
lines changed Original file line number Diff line number Diff line change @@ -114,3 +114,16 @@ def check_status(self, job):
114
114
return JobStatus .terminated
115
115
else :
116
116
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
You can’t perform that action at this time.
0 commit comments