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):
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
You can’t perform that action at this time.
0 commit comments