| [1] | 1 | %{ |
|---|
| [12] | 2 | /* $Id$ */ |
|---|
| [1] | 3 | /* |
|---|
| 4 | * FedStage DRMAA for PBS Pro |
|---|
| 5 | * Copyright (C) 2006-2009 FedStage Systems |
|---|
| [26] | 6 | * Copyright (C) 2011 Poznan Supercomputing and Networking Center |
|---|
| [1] | 7 | * This program is free software: you can redistribute it and/or modify |
|---|
| 8 | * it under the terms of the GNU General Public License as published by |
|---|
| 9 | * the Free Software Foundation, either version 3 of the License, or |
|---|
| 10 | * (at your option) any later version. |
|---|
| 11 | * |
|---|
| 12 | * This program is distributed in the hope that it will be useful, |
|---|
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
|---|
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|---|
| 15 | * GNU General Public License for more details. |
|---|
| 16 | * |
|---|
| 17 | * You should have received a copy of the GNU General Public License |
|---|
| 18 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
|---|
| 19 | */ |
|---|
| 20 | #ifdef HAVE_CONFIG_H |
|---|
| 21 | # include <config.h> |
|---|
| 22 | #endif |
|---|
| 23 | |
|---|
| 24 | #include <drmaa_utils/drmaa_attrib.h> |
|---|
| 25 | #include <drmaa_utils/template.h> |
|---|
| 26 | #include <pbs_drmaa/pbs_attrib.h> |
|---|
| 27 | |
|---|
| 28 | #ifndef lint |
|---|
| 29 | static char rcsid[] |
|---|
| 30 | # ifdef __GNUC__ |
|---|
| 31 | __attribute__ ((unused)) |
|---|
| 32 | # endif |
|---|
| [12] | 33 | = "$Id$"; |
|---|
| [1] | 34 | #endif |
|---|
| 35 | |
|---|
| 36 | extern const fsd_attribute_t pbsdrmaa_pbs_attributes[]; |
|---|
| 37 | #define t(code) \ |
|---|
| 38 | ( & pbsdrmaa_pbs_attributes[ code ] ) |
|---|
| 39 | %} |
|---|
| 40 | |
|---|
| 41 | %language=ANSI-C |
|---|
| 42 | %includes |
|---|
| 43 | %readonly-tables |
|---|
| 44 | %pic |
|---|
| 45 | %define lookup-function-name pbsdrmaa_pbs_attrib_lookup |
|---|
| 46 | %struct-type |
|---|
| 47 | struct pbs_attrib { int name; const fsd_attribute_t *attr; } |
|---|
| 48 | |
|---|
| 49 | %% |
|---|
| 50 | Execution_Time, t(PBSDRMAA_ATTR_EXECUTION_TIME) |
|---|
| 51 | Checkpoint, t(PBSDRMAA_ATTR_CHECKPOINT) |
|---|
| 52 | Error_Path, t(PBSDRMAA_ATTR_ERROR_PATH) |
|---|
| 53 | group_list, t(PBSDRMAA_ATTR_GROUPS) |
|---|
| 54 | Hold_Types, t(PBSDRMAA_ATTR_HOLD_TYPES) |
|---|
| 55 | Join_Path, t(PBSDRMAA_ATTR_JOIN_FILES) |
|---|
| 56 | Keep_Files, t(PBSDRMAA_ATTR_KEEP_FILES) |
|---|
| 57 | # Resource_List, t(PBSDRMAA_ATTR_RESOURCES) |
|---|
| 58 | Resource_List.cput, t(PBSDRMAA_ATTR_CPU_TIME_LIMIT) |
|---|
| 59 | Resource_List.file, t(PBSDRMAA_ATTR_FILE_SIZE_LIMIT) |
|---|
| 60 | Resource_List.nice, t(PBSDRMAA_ATTR_NICE) |
|---|
| 61 | Resource_List.vmem, t(PBSDRMAA_ATTR_VMEM_LIMIT) |
|---|
| [4] | 62 | Resource_List.mem, t(PBSDRMAA_ATTR_MEM_LIMIT) |
|---|
| [1] | 63 | Resource_List.pcput, t(PBSDRMAA_ATTR_SINGLE_CPU_TIME_LIMIT) |
|---|
| 64 | Resource_List.pmem, t(PBSDRMAA_ATTR_SINGLE_RSS_LIMIT) |
|---|
| 65 | Resource_List.pvmem, t(PBSDRMAA_ATTR_SINGLE_VMEM_LIMIT) |
|---|
| 66 | Resource_List.walltime, t(PBSDRMAA_ATTR_WALLTIME_LIMIT) |
|---|
| 67 | Resource_List.arch, t(PBSDRMAA_ATTR_ARCHITECTURE) |
|---|
| 68 | Resource_List.host, t(PBSDRMAA_ATTR_HOST) |
|---|
| 69 | Resource_List.nodes, t(PBSDRMAA_ATTR_NODES) |
|---|
| [42] | 70 | Resource_List.procs, t(PBSDRMAA_ATTR_PROCS) |
|---|
| [44] | 71 | Resource_List.ncpus, t(PBSDRMAA_ATTR_NCPUS) |
|---|
| [47] | 72 | Resource_List.software, t(PBSDRMAA_ATTR_SOFTWARE) |
|---|
| [1] | 73 | Mail_Points, t(PBSDRMAA_ATTR_MAIL_POINTS) |
|---|
| 74 | Output_Path, t(PBSDRMAA_ATTR_OUTPUT_PATH) |
|---|
| 75 | Priority, t(PBSDRMAA_ATTR_PRIORITY) |
|---|
| 76 | destination, t(PBSDRMAA_ATTR_DESITINATION) |
|---|
| 77 | Rerunable, t(PBSDRMAA_ATTR_RERUNABLE) |
|---|
| 78 | job_array_request, t(PBSDRMAA_ATTR_BULK_REQ) |
|---|
| 79 | job_array_id, t(PBSDRMAA_ATTR_BULK_IDX) |
|---|
| 80 | User_List, t(PBSDRMAA_ATTR_USERS) |
|---|
| 81 | Variable_List, t(PBSDRMAA_ATTR_JOB_ENVIRONMENT) |
|---|
| 82 | Account_Name, t(PBSDRMAA_ATTR_ACCOUNT_NAME) |
|---|
| 83 | Mail_Users, t(PBSDRMAA_ATTR_EMAIL) |
|---|
| 84 | Job_Name, t(PBSDRMAA_ATTR_JOB_NAME) |
|---|
| 85 | Shell_Path_List, t(PBSDRMAA_ATTR_SHELL) |
|---|
| 86 | depend, t(PBSDRMAA_ATTR_DEPEND) |
|---|
| 87 | interactive, t(PBSDRMAA_ATTR_INTERACTIVE) |
|---|
| 88 | stagein, t(PBSDRMAA_ATTR_STAGEIN) |
|---|
| 89 | stageout, t(PBSDRMAA_ATTR_STAGEOUT) |
|---|
| 90 | # additional job and general attribute names |
|---|
| 91 | exec_host, t(PBSDRMAA_ATTR_EXECUTION_HOST) |
|---|
| [26] | 92 | exec_vnode, t(PBSDRMAA_ATTR_EXECUTION_VNODE) |
|---|
| [1] | 93 | Job_Owner, t(PBSDRMAA_ATTR_JOB_OWNER) |
|---|
| 94 | resources_used, t(PBSDRMAA_ATTR_RESOURCES_USED) |
|---|
| 95 | job_state, t(PBSDRMAA_ATTR_JOB_STATE) |
|---|
| 96 | queue, t(PBSDRMAA_ATTR_QUEUE) |
|---|
| 97 | server, t(PBSDRMAA_ATTR_SERVER) |
|---|
| 98 | comment, t(PBSDRMAA_ATTR_COMMENT) |
|---|
| 99 | exit_status, t(PBSDRMAA_ATTR_EXIT_STATUS) |
|---|
| 100 | start_time, t(PBSDRMAA_ATTR_START_TIME) |
|---|
| 101 | x, t(PBSDRMAA_ATTR_EXTENSION) |
|---|
| 102 | submit_args, t(PBSDRMAA_ATTR_SUBMIT_ARGS) |
|---|
| 103 | mtime, t(PBSDRMAA_ATTR_MTIME) |
|---|
| [49] | 104 | Resource_List.node_properties, t(PBSDRMAA_ATTR_NODE_PROPERTIES) |
|---|
| 105 | Resource_List.custom_resources, t(PBSDRMAA_ATTR_CUSTOM_RESOURCES) |
|---|
| [1] | 106 | %% |
|---|
| 107 | #undef t |
|---|
| 108 | |
|---|
| 109 | int |
|---|
| 110 | pbsdrmaa_pbs_attrib_by_name( const char *name ) |
|---|
| 111 | { |
|---|
| 112 | const struct pbs_attrib *found; |
|---|
| 113 | found = pbsdrmaa_pbs_attrib_lookup( name, strlen(name) ); |
|---|
| 114 | if( found ) |
|---|
| 115 | return found->attr->code; |
|---|
| 116 | else |
|---|
| 117 | return -1; |
|---|
| 118 | } |
|---|
| 119 | |
|---|
| 120 | static const fsd_attribute_t * |
|---|
| 121 | pbsdrmaa_pbs_template_by_name( const fsd_template_t *self, const char *name ) |
|---|
| 122 | { |
|---|
| 123 | const struct pbs_attrib *found; |
|---|
| 124 | found = pbsdrmaa_pbs_attrib_lookup( name, strlen(name) ); |
|---|
| 125 | if( found != NULL ) |
|---|
| 126 | { |
|---|
| 127 | fsd_assert( found->attr - pbsdrmaa_pbs_attributes == found->attr->code ); |
|---|
| 128 | return found->attr; |
|---|
| 129 | } |
|---|
| 130 | else |
|---|
| 131 | fsd_exc_raise_fmt( |
|---|
| 132 | FSD_ERRNO_INVALID_ARGUMENT, |
|---|
| 133 | "invalid PBS attribute name: %s", name |
|---|
| 134 | ); |
|---|
| 135 | } |
|---|
| 136 | |
|---|
| 137 | |
|---|
| 138 | static const fsd_attribute_t * |
|---|
| 139 | pbsdrmaa_pbs_template_by_code( const fsd_template_t *self, int code ) |
|---|
| 140 | { |
|---|
| 141 | if( 0 <= code && code < PBSDRMAA_N_PBS_ATTRIBUTES ) |
|---|
| 142 | { |
|---|
| 143 | fsd_assert( pbsdrmaa_pbs_attributes[code].code == code ); |
|---|
| 144 | return & pbsdrmaa_pbs_attributes[ code ]; |
|---|
| 145 | } |
|---|
| 146 | else |
|---|
| 147 | fsd_exc_raise_fmt( |
|---|
| 148 | FSD_ERRNO_INVALID_ARGUMENT, |
|---|
| 149 | "invalid PBS attribute code: %d", code |
|---|
| 150 | ); |
|---|
| 151 | } |
|---|
| 152 | |
|---|
| 153 | |
|---|
| 154 | fsd_template_t * |
|---|
| 155 | pbsdrmaa_pbs_template_new(void) |
|---|
| 156 | { |
|---|
| 157 | return fsd_template_new( |
|---|
| 158 | pbsdrmaa_pbs_template_by_name, |
|---|
| 159 | pbsdrmaa_pbs_template_by_code, |
|---|
| 160 | PBSDRMAA_N_PBS_ATTRIBUTES |
|---|
| 161 | ); |
|---|
| 162 | } |
|---|
| 163 | |
|---|
| 164 | const fsd_attribute_t pbsdrmaa_pbs_attributes[ PBSDRMAA_N_PBS_ATTRIBUTES ] = { |
|---|
| 165 | { "Execution_Time", PBSDRMAA_ATTR_EXECUTION_TIME, false }, |
|---|
| 166 | { "Checkpoint", PBSDRMAA_ATTR_CHECKPOINT, false }, |
|---|
| 167 | { "Error_Path", PBSDRMAA_ATTR_ERROR_PATH, false }, |
|---|
| 168 | { "group_list", PBSDRMAA_ATTR_GROUPS, false }, |
|---|
| 169 | { "Hold_Types", PBSDRMAA_ATTR_HOLD_TYPES, false }, |
|---|
| 170 | { "Join_Path", PBSDRMAA_ATTR_JOIN_FILES, false }, |
|---|
| 171 | { "Keep_Files", PBSDRMAA_ATTR_KEEP_FILES, false }, |
|---|
| 172 | { "Resource_List.cput", PBSDRMAA_ATTR_CPU_TIME_LIMIT, false }, |
|---|
| 173 | { "Resource_List.file", PBSDRMAA_ATTR_FILE_SIZE_LIMIT, false }, |
|---|
| 174 | { "Resource_List.nice", PBSDRMAA_ATTR_NICE, false }, |
|---|
| 175 | { "Resource_List.vmem", PBSDRMAA_ATTR_VMEM_LIMIT, false }, |
|---|
| [4] | 176 | { "Resource_List.mem", PBSDRMAA_ATTR_MEM_LIMIT, false }, |
|---|
| [1] | 177 | { "Resource_List.pcput", PBSDRMAA_ATTR_SINGLE_CPU_TIME_LIMIT, false }, |
|---|
| 178 | { "Resource_List.pmem", PBSDRMAA_ATTR_SINGLE_RSS_LIMIT, false }, |
|---|
| 179 | { "Resource_List.pvmem", PBSDRMAA_ATTR_SINGLE_VMEM_LIMIT, false }, |
|---|
| 180 | { "Resource_List.walltime", PBSDRMAA_ATTR_WALLTIME_LIMIT, false }, |
|---|
| 181 | { "Resource_List.arch", PBSDRMAA_ATTR_ARCHITECTURE, false }, |
|---|
| 182 | { "Resource_List.host", PBSDRMAA_ATTR_HOST, false }, |
|---|
| 183 | { "Resource_List.nodes", PBSDRMAA_ATTR_NODES, false }, |
|---|
| [42] | 184 | { "Resource_List.procs", PBSDRMAA_ATTR_PROCS, false }, |
|---|
| [43] | 185 | { "Resource_List.ncpus", PBSDRMAA_ATTR_NCPUS, false }, |
|---|
| [47] | 186 | { "Resource_List.software", PBSDRMAA_ATTR_SOFTWARE, false }, |
|---|
| [1] | 187 | { "Mail_Points", PBSDRMAA_ATTR_MAIL_POINTS, false }, |
|---|
| 188 | { "Output_Path", PBSDRMAA_ATTR_OUTPUT_PATH, false }, |
|---|
| 189 | { "Priority", PBSDRMAA_ATTR_PRIORITY, false }, |
|---|
| 190 | { "destination", PBSDRMAA_ATTR_DESITINATION, false }, |
|---|
| 191 | { "Rerunable", PBSDRMAA_ATTR_RERUNABLE, false }, |
|---|
| 192 | { "job_array_request", PBSDRMAA_ATTR_BULK_REQ, false }, |
|---|
| 193 | { "job_array_id", PBSDRMAA_ATTR_BULK_IDX, false }, |
|---|
| 194 | { "User_List", PBSDRMAA_ATTR_USERS, false }, |
|---|
| 195 | { "Variable_List", PBSDRMAA_ATTR_JOB_ENVIRONMENT, false }, |
|---|
| 196 | { "Account_Name", PBSDRMAA_ATTR_ACCOUNT_NAME, false }, |
|---|
| 197 | { "Mail_Users", PBSDRMAA_ATTR_EMAIL, false }, |
|---|
| 198 | { "Job_Name", PBSDRMAA_ATTR_JOB_NAME, false }, |
|---|
| 199 | { "Shell_Path_List", PBSDRMAA_ATTR_SHELL, false }, |
|---|
| 200 | { "depend", PBSDRMAA_ATTR_DEPEND, false }, |
|---|
| 201 | { "interactive", PBSDRMAA_ATTR_INTERACTIVE, false }, |
|---|
| 202 | { "stagein", PBSDRMAA_ATTR_STAGEIN, false }, |
|---|
| 203 | { "stageout", PBSDRMAA_ATTR_STAGEOUT, false }, |
|---|
| 204 | { "exec_host", PBSDRMAA_ATTR_EXECUTION_HOST, false }, |
|---|
| [26] | 205 | { "exec_vnode", PBSDRMAA_ATTR_EXECUTION_VNODE, false }, |
|---|
| [1] | 206 | { "Job_Owner", PBSDRMAA_ATTR_JOB_OWNER, false }, |
|---|
| 207 | { "resources_used", PBSDRMAA_ATTR_RESOURCES_USED, false }, |
|---|
| 208 | { "job_state", PBSDRMAA_ATTR_JOB_STATE, false }, |
|---|
| 209 | { "queue", PBSDRMAA_ATTR_QUEUE, false }, |
|---|
| 210 | { "server", PBSDRMAA_ATTR_SERVER, false }, |
|---|
| 211 | { "comment", PBSDRMAA_ATTR_COMMENT, false }, |
|---|
| 212 | { "exit_status", PBSDRMAA_ATTR_EXIT_STATUS, false }, |
|---|
| 213 | { "start_time", PBSDRMAA_ATTR_START_TIME, false }, |
|---|
| 214 | { "x", PBSDRMAA_ATTR_EXTENSION, false }, |
|---|
| 215 | { "submit_args", PBSDRMAA_ATTR_SUBMIT_ARGS, false }, |
|---|
| 216 | { "mtime", PBSDRMAA_ATTR_MTIME, false }, |
|---|
| [49] | 217 | { "Resource_List.node_properties", PBSDRMAA_ATTR_NODE_PROPERTIES, false }, /* DRMAA extenstions */ |
|---|
| 218 | { "Resource_List.custom_resources", PBSDRMAA_ATTR_CUSTOM_RESOURCES, false }, |
|---|
| [48] | 219 | |
|---|
| [1] | 220 | }; |
|---|
| 221 | |
|---|
| 222 | /* vim: set ft=c: */ |
|---|
