[1] | 1 | /* |
---|
| 2 | * PSNC DRMAA for SLURM |
---|
[13] | 3 | * Copyright (C) 2011 Poznan Supercomputing and Networking Center |
---|
[1] | 4 | * |
---|
| 5 | * This program is free software: you can redistribute it and/or modify |
---|
| 6 | * it under the terms of the GNU General Public License as published by |
---|
| 7 | * the Free Software Foundation, either version 3 of the License, or |
---|
| 8 | * (at your option) any later version. |
---|
| 9 | * |
---|
| 10 | * This program is distributed in the hope that it will be useful, |
---|
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
| 13 | * GNU General Public License for more details. |
---|
| 14 | * |
---|
| 15 | * You should have received a copy of the GNU General Public License |
---|
| 16 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
---|
| 17 | */ |
---|
| 18 | |
---|
| 19 | #include <drmaa_utils/common.h> |
---|
| 20 | #include <drmaa_utils/exception.h> |
---|
| 21 | #include <slurm_drmaa/util.h> |
---|
| 22 | #include <string.h> |
---|
| 23 | |
---|
| 24 | #include <time.h> |
---|
| 25 | #include <drmaa_utils/datetime.h> |
---|
| 26 | #include <drmaa_utils/datetime_impl.h> |
---|
| 27 | |
---|
| 28 | #ifndef lint |
---|
| 29 | static char rcsid[] |
---|
| 30 | # ifdef __GNUC__ |
---|
| 31 | __attribute__ ((unused)) |
---|
| 32 | # endif |
---|
[5] | 33 | = "$Id$"; |
---|
[1] | 34 | #endif |
---|
| 35 | |
---|
[24] | 36 | unsigned int |
---|
[1] | 37 | slurmdrmaa_datetime_parse( const char *string ) |
---|
| 38 | { |
---|
| 39 | #ifdef DEBUGGING |
---|
| 40 | char dbg[256]; |
---|
| 41 | #endif |
---|
| 42 | fsd_dt_parser_t *volatile parser = NULL; |
---|
| 43 | fsd_dt_lexer_t *volatile lexer = NULL; |
---|
| 44 | int parse_err = 0; |
---|
| 45 | fsd_datetime_t dt; |
---|
| 46 | |
---|
| 47 | memset(&dt, 0, sizeof(fsd_datetime_t)); |
---|
| 48 | |
---|
| 49 | fsd_log_enter(( "(%s)", string )); |
---|
| 50 | TRY |
---|
| 51 | { |
---|
| 52 | fsd_malloc( parser, fsd_dt_parser_t ); |
---|
| 53 | fsd_malloc( lexer, fsd_dt_lexer_t ); |
---|
| 54 | |
---|
| 55 | parser->lexer = lexer; |
---|
| 56 | parser->n_errors = 0; |
---|
| 57 | lexer->parser = parser; |
---|
| 58 | lexer->begin = lexer->p = (unsigned char*)string; |
---|
| 59 | lexer->end = (unsigned char*)( string + strlen(string) ); |
---|
| 60 | parse_err = fsd_dt_parse( parser, lexer ); |
---|
| 61 | if( parse_err || parser->n_errors ) |
---|
| 62 | fsd_exc_raise_fmt( |
---|
| 63 | FSD_ERRNO_INVALID_VALUE_FORMAT, |
---|
| 64 | "invalid date/time format: %s", string |
---|
| 65 | ); |
---|
| 66 | |
---|
| 67 | dt = parser->result; |
---|
| 68 | #ifdef DEBUGGING |
---|
| 69 | fsd_datetime_dump( &dt, dbg, sizeof(dbg) ); |
---|
| 70 | fsd_log_debug(( "parsed: %s", dbg )); |
---|
| 71 | #endif |
---|
| 72 | } |
---|
| 73 | FINALLY |
---|
| 74 | { |
---|
| 75 | fsd_free( parser ); |
---|
| 76 | fsd_free( lexer ); |
---|
| 77 | } |
---|
| 78 | END_TRY |
---|
| 79 | |
---|
[24] | 80 | fsd_log_return(( "(%s) =%u minutes", string, (unsigned)60*dt.hour+dt.minute )); |
---|
[1] | 81 | return 60*dt.hour+dt.minute; |
---|
| 82 | } |
---|
| 83 | |
---|
[28] | 84 | /* taken from SLURM src/common/proc_args.c */ |
---|
| 85 | static int slurmdrmaa_mail_type_parse(const char *mail_type_str) |
---|
| 86 | { |
---|
| 87 | int rc = 0; |
---|
| 88 | |
---|
| 89 | |
---|
| 90 | if (strcasecmp(mail_type_str, "BEGIN") == 0) |
---|
| 91 | rc = MAIL_JOB_BEGIN; |
---|
| 92 | else if (strcasecmp(mail_type_str, "END") == 0) |
---|
| 93 | rc = MAIL_JOB_END; |
---|
| 94 | else if (strcasecmp(mail_type_str, "FAIL") == 0) |
---|
| 95 | rc = MAIL_JOB_FAIL; |
---|
[29] | 96 | #if SLURM_VERSION_NUMBER >= SLURM_VERSION_NUM(2,2,0) |
---|
[28] | 97 | else if (strcasecmp(mail_type_str, "REQUEUE") == 0) |
---|
| 98 | rc = MAIL_JOB_REQUEUE; |
---|
[29] | 99 | #endif |
---|
[28] | 100 | else if (strcasecmp(mail_type_str, "ALL") == 0) |
---|
[29] | 101 | #if SLURM_VERSION_NUMBER >= SLURM_VERSION_NUM(2,2,0) |
---|
[28] | 102 | rc = MAIL_JOB_BEGIN | MAIL_JOB_END | MAIL_JOB_FAIL | MAIL_JOB_REQUEUE; |
---|
[29] | 103 | #else |
---|
| 104 | rc = MAIL_JOB_BEGIN | MAIL_JOB_END | MAIL_JOB_FAIL; |
---|
| 105 | #endif |
---|
| 106 | else { |
---|
| 107 | fsd_log_error(("Unknown mail type: %s", mail_type_str)); |
---|
| 108 | } |
---|
[28] | 109 | |
---|
| 110 | return rc; |
---|
| 111 | } |
---|
| 112 | |
---|
| 113 | |
---|
[1] | 114 | enum slurm_native { |
---|
| 115 | SLURM_NATIVE_ACCOUNT, |
---|
| 116 | SLURM_NATIVE_ACCTG_FREQ, |
---|
| 117 | SLURM_NATIVE_COMMENT, |
---|
| 118 | SLURM_NATIVE_CONSTRAINT, |
---|
| 119 | SLURM_NATIVE_CONTIGUOUS, |
---|
| 120 | SLURM_NATIVE_EXCLUSIVE, |
---|
| 121 | SLURM_NATIVE_MEM, |
---|
| 122 | SLURM_NATIVE_MEM_PER_CPU, |
---|
| 123 | SLURM_NATIVE_MINCPUS, |
---|
| 124 | SLURM_NATIVE_NODELIST, |
---|
| 125 | SLURM_NATIVE_NODES, |
---|
| 126 | SLURM_NATIVE_NTASKS_PER_NODE, |
---|
| 127 | SLURM_NATIVE_PARTITION, |
---|
| 128 | SLURM_NATIVE_QOS, |
---|
| 129 | SLURM_NATIVE_REQUEUE, |
---|
| 130 | SLURM_NATIVE_RESERVATION, |
---|
[18] | 131 | SLURM_NATIVE_SHARE, |
---|
[19] | 132 | SLURM_NATIVE_JOB_NAME, |
---|
[24] | 133 | SLURM_NATIVE_TIME_LIMIT, |
---|
[28] | 134 | SLURM_NATIVE_NTASKS, |
---|
| 135 | SLURM_NATIVE_GRES, |
---|
| 136 | SLURM_NATIVE_NO_KILL, |
---|
| 137 | SLURM_NATIVE_LICENSES, |
---|
| 138 | SLURM_NATIVE_MAIL_TYPE, |
---|
| 139 | SLURM_NATIVE_NO_REQUEUE, |
---|
| 140 | SLURM_NATIVE_EXCLUDE |
---|
[1] | 141 | }; |
---|
| 142 | |
---|
| 143 | void |
---|
| 144 | slurmdrmaa_init_job_desc(job_desc_msg_t *job_desc) |
---|
| 145 | { |
---|
| 146 | memset(job_desc, 0, sizeof(job_desc_msg_t)); |
---|
| 147 | } |
---|
| 148 | |
---|
| 149 | void |
---|
| 150 | slurmdrmaa_free_job_desc(job_desc_msg_t *job_desc) |
---|
| 151 | { |
---|
| 152 | unsigned i = 0; |
---|
| 153 | fsd_log_enter(( "" )); |
---|
| 154 | fsd_free(job_desc->account); |
---|
| 155 | fsd_free(job_desc->comment); |
---|
| 156 | |
---|
| 157 | for(i = 0;i<job_desc->env_size;i++) |
---|
| 158 | { |
---|
| 159 | fsd_free(job_desc->environment[i]); |
---|
| 160 | } |
---|
| 161 | fsd_free(job_desc->environment); |
---|
| 162 | fsd_free(job_desc->features); |
---|
| 163 | fsd_free(job_desc->name); |
---|
| 164 | fsd_free(job_desc->mail_user); |
---|
| 165 | fsd_free(job_desc->partition); |
---|
| 166 | fsd_free(job_desc->qos); |
---|
| 167 | fsd_free(job_desc->script); |
---|
| 168 | fsd_free(job_desc->std_in); |
---|
| 169 | fsd_free(job_desc->std_out); |
---|
| 170 | fsd_free(job_desc->std_err); |
---|
| 171 | fsd_free(job_desc->work_dir); |
---|
[29] | 172 | fsd_free(job_desc->exc_nodes); |
---|
| 173 | #if SLURM_VERSION_NUMBER >= SLURM_VERSION_NUM(2,2,0) |
---|
[28] | 174 | fsd_free(job_desc->gres); |
---|
[29] | 175 | #endif |
---|
[1] | 176 | |
---|
| 177 | fsd_log_return(( "" )); |
---|
| 178 | } |
---|
| 179 | |
---|
| 180 | void |
---|
| 181 | slurmdrmaa_add_attribute(job_desc_msg_t *job_desc, unsigned attr, const char *value) |
---|
| 182 | { |
---|
| 183 | char * ptr = NULL; |
---|
| 184 | char * rest = NULL; |
---|
| 185 | char * token = NULL; |
---|
| 186 | |
---|
| 187 | switch(attr) |
---|
| 188 | { |
---|
| 189 | case SLURM_NATIVE_ACCOUNT: |
---|
| 190 | fsd_free(job_desc->account); |
---|
| 191 | fsd_log_debug(("# account = %s",value)); |
---|
| 192 | job_desc->account = fsd_strdup(value); |
---|
| 193 | break; |
---|
| 194 | case SLURM_NATIVE_ACCTG_FREQ: |
---|
| 195 | fsd_log_debug(("# acctg_freq = %s",value)); |
---|
| 196 | job_desc->acctg_freq = fsd_atoi(value); |
---|
| 197 | break; |
---|
| 198 | case SLURM_NATIVE_COMMENT: |
---|
| 199 | fsd_free(job_desc->comment); |
---|
| 200 | fsd_log_debug(("# comment = %s",value)); |
---|
| 201 | job_desc->comment = fsd_strdup(value); |
---|
| 202 | break; |
---|
| 203 | case SLURM_NATIVE_CONSTRAINT: |
---|
| 204 | fsd_free(job_desc->features); |
---|
| 205 | fsd_log_debug(("# constraints -> features = %s",value)); |
---|
| 206 | job_desc->features = fsd_strdup(value); |
---|
| 207 | break; |
---|
| 208 | case SLURM_NATIVE_CONTIGUOUS: |
---|
| 209 | fsd_log_debug(( "# contiguous = 1")); |
---|
| 210 | job_desc->contiguous = 1; |
---|
| 211 | break; |
---|
| 212 | case SLURM_NATIVE_EXCLUSIVE: |
---|
| 213 | fsd_log_debug(( "# exclusive -> shared = 0")); |
---|
| 214 | job_desc->shared = 0; |
---|
| 215 | break; |
---|
| 216 | case SLURM_NATIVE_MEM: |
---|
[13] | 217 | #if SLURM_VERSION_NUMBER < SLURM_VERSION_NUM(2,2,0) |
---|
[1] | 218 | if(job_desc->job_min_memory == NO_VAL || fsd_atoi(value) > (int)job_desc->job_min_memory) { |
---|
| 219 | fsd_log_debug(("# job_min_memory = %s",value)); |
---|
| 220 | job_desc->job_min_memory = fsd_atoi(value); |
---|
| 221 | } |
---|
[13] | 222 | #else |
---|
| 223 | if(job_desc->pn_min_memory == NO_VAL || fsd_atoi(value) > (int)job_desc->pn_min_memory) { |
---|
| 224 | fsd_log_debug(("# pn_min_memory = %s",value)); |
---|
| 225 | job_desc->pn_min_memory = fsd_atoi(value); |
---|
| 226 | } |
---|
| 227 | #endif |
---|
| 228 | else { |
---|
[1] | 229 | fsd_log_debug(("mem value defined lower or equal to mem-per-cpu or value defined before")); |
---|
| 230 | } |
---|
| 231 | break; |
---|
| 232 | case SLURM_NATIVE_MEM_PER_CPU: |
---|
[13] | 233 | #if SLURM_VERSION_NUMBER < SLURM_VERSION_NUM(2,2,0) |
---|
[1] | 234 | if(job_desc->job_min_memory == NO_VAL || fsd_atoi(value) > (int)job_desc->job_min_memory) { |
---|
| 235 | fsd_log_debug(("# job_min_memory = %s",value)); |
---|
| 236 | job_desc->job_min_memory = fsd_atoi(value); |
---|
| 237 | } |
---|
[13] | 238 | #else |
---|
| 239 | if(job_desc->pn_min_memory == NO_VAL || fsd_atoi(value) > (int)job_desc->pn_min_memory) { |
---|
| 240 | fsd_log_debug(("# pn_min_memory = %s",value)); |
---|
| 241 | job_desc->pn_min_memory = fsd_atoi(value); |
---|
| 242 | } |
---|
| 243 | #endif |
---|
| 244 | else { |
---|
[1] | 245 | fsd_log_debug(("mem-per-cpu value defined lower or equal to mem or value defined before")); |
---|
| 246 | } |
---|
| 247 | break; |
---|
| 248 | case SLURM_NATIVE_MINCPUS: |
---|
[13] | 249 | #if SLURM_VERSION_NUMBER < SLURM_VERSION_NUM(2,2,0) |
---|
[1] | 250 | fsd_log_debug(("# job_min_cpus = %s",value)); |
---|
| 251 | job_desc->job_min_cpus = fsd_atoi(value); |
---|
[13] | 252 | #else |
---|
| 253 | fsd_log_debug(("# min_cpus = %s",value)); |
---|
| 254 | job_desc->min_cpus = fsd_atoi(value); |
---|
| 255 | #endif |
---|
[1] | 256 | break; |
---|
| 257 | case SLURM_NATIVE_NODELIST: |
---|
| 258 | fsd_free(job_desc->req_nodes); |
---|
| 259 | fsd_log_debug(("# node-list - > req_nodes = %s",value)); |
---|
| 260 | job_desc->req_nodes = fsd_strdup(value); |
---|
| 261 | break; |
---|
| 262 | case SLURM_NATIVE_NODES: |
---|
| 263 | ptr = strdup(value); |
---|
[24] | 264 | |
---|
[1] | 265 | if((token = strtok_r(ptr,"=",&rest)) == NULL){ |
---|
| 266 | fsd_log_error(("strtok_r returned NULL")); |
---|
| 267 | } |
---|
| 268 | |
---|
| 269 | fsd_log_debug(("nodes: %s ->",value)); |
---|
| 270 | fsd_log_debug(("# min_nodes = %s",token)); |
---|
| 271 | job_desc->min_nodes = fsd_atoi(token); |
---|
| 272 | |
---|
| 273 | if(strcmp(rest,"") !=0 ) { |
---|
| 274 | fsd_log_debug(("# max_nodes = %s",rest)); |
---|
| 275 | job_desc->max_nodes = fsd_atoi(rest); |
---|
| 276 | } |
---|
| 277 | fsd_free(ptr); |
---|
| 278 | break; |
---|
| 279 | case SLURM_NATIVE_NTASKS_PER_NODE: |
---|
| 280 | fsd_log_debug(("# ntasks_per_node = %s",value)); |
---|
| 281 | job_desc->ntasks_per_node = fsd_atoi(value); |
---|
| 282 | break; |
---|
| 283 | case SLURM_NATIVE_PARTITION: |
---|
| 284 | fsd_free(job_desc->partition); |
---|
| 285 | fsd_log_debug(("# partition = %s",value)); |
---|
| 286 | job_desc->partition = fsd_strdup(value); |
---|
| 287 | break; |
---|
| 288 | case SLURM_NATIVE_QOS: |
---|
| 289 | fsd_free(job_desc->qos); |
---|
| 290 | fsd_log_debug(("# qos = %s",value)); |
---|
| 291 | job_desc->qos = fsd_strdup(value); |
---|
| 292 | break; |
---|
| 293 | case SLURM_NATIVE_REQUEUE: |
---|
| 294 | fsd_log_debug(( "# requeue = 1")); |
---|
| 295 | job_desc->requeue = 1; |
---|
| 296 | break; |
---|
| 297 | case SLURM_NATIVE_RESERVATION: |
---|
| 298 | fsd_log_debug(("# reservation = %s",value)); |
---|
| 299 | job_desc->reservation = fsd_strdup(value); |
---|
| 300 | break; |
---|
| 301 | case SLURM_NATIVE_SHARE: |
---|
| 302 | fsd_log_debug(("# shared = 1")); |
---|
| 303 | job_desc->shared = 1; |
---|
| 304 | break; |
---|
[18] | 305 | case SLURM_NATIVE_JOB_NAME: |
---|
[24] | 306 | fsd_log_debug(("# job_name = %s",job_desc->name)); |
---|
| 307 | job_desc->name = fsd_strdup(value); |
---|
[18] | 308 | break; |
---|
[24] | 309 | case SLURM_NATIVE_NTASKS: |
---|
| 310 | fsd_log_debug(("# ntasks = %s",value)); |
---|
| 311 | job_desc->num_tasks = fsd_atoi(value); |
---|
| 312 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MINCPUS,value); |
---|
| 313 | break; |
---|
| 314 | case SLURM_NATIVE_TIME_LIMIT: |
---|
| 315 | fsd_log_debug(("# time_limit = %s",value)); |
---|
| 316 | job_desc->time_limit = slurmdrmaa_datetime_parse(value); |
---|
| 317 | break; |
---|
[28] | 318 | case SLURM_NATIVE_GRES: |
---|
[29] | 319 | #if SLURM_VERSION_NUMBER >= SLURM_VERSION_NUM(2,2,0) |
---|
[28] | 320 | fsd_log_debug(("# gres = %s",value)); |
---|
| 321 | job_desc->gres = fsd_strdup(value); |
---|
[29] | 322 | #else |
---|
| 323 | fsd_log_error(("GRES not supported in this version of SLURM.")); |
---|
| 324 | #endif |
---|
[28] | 325 | break; |
---|
| 326 | case SLURM_NATIVE_NO_KILL: |
---|
| 327 | fsd_log_debug(("# no_kill = 1")); |
---|
| 328 | job_desc->kill_on_node_fail = 0; |
---|
| 329 | break; |
---|
| 330 | case SLURM_NATIVE_LICENSES: |
---|
| 331 | fsd_log_debug(("# licenses = %s", value)); |
---|
| 332 | job_desc->licenses = fsd_strdup(value); |
---|
| 333 | break; |
---|
| 334 | case SLURM_NATIVE_MAIL_TYPE: |
---|
| 335 | fsd_log_debug(("# mail_type = %s", value)); |
---|
| 336 | job_desc->mail_type = slurmdrmaa_mail_type_parse(value); |
---|
| 337 | break; |
---|
| 338 | case SLURM_NATIVE_NO_REQUEUE: |
---|
| 339 | fsd_log_debug(("# requeue = 0")); |
---|
| 340 | job_desc->requeue = 0; |
---|
| 341 | break; |
---|
| 342 | case SLURM_NATIVE_EXCLUDE: |
---|
| 343 | fsd_log_debug(("# exclude = %s", value)); |
---|
| 344 | job_desc->exc_nodes = fsd_strdup(value); |
---|
| 345 | break; |
---|
[24] | 346 | |
---|
[1] | 347 | default: |
---|
| 348 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE,"Invalid attribute"); |
---|
| 349 | } |
---|
| 350 | } |
---|
| 351 | |
---|
| 352 | void |
---|
| 353 | slurmdrmaa_parse_additional_attr(job_desc_msg_t *job_desc,const char *add_attr) |
---|
| 354 | { |
---|
| 355 | char * volatile name = NULL; |
---|
| 356 | char *value = NULL; |
---|
| 357 | char *ctxt = NULL; |
---|
| 358 | char * volatile add_attr_copy = fsd_strdup(add_attr); |
---|
| 359 | |
---|
| 360 | TRY |
---|
| 361 | { |
---|
| 362 | name = fsd_strdup(strtok_r(add_attr_copy, "=", &ctxt)); |
---|
| 363 | value = strtok_r(NULL, "=", &ctxt); |
---|
[28] | 364 | /* |
---|
| 365 | * TODO: move it to slurmdrmaa_add_attribute |
---|
| 366 | if (value == NULL) { |
---|
[24] | 367 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE, |
---|
| 368 | "Invalid native specification: %s Missing '='.", add_attr_copy); |
---|
[28] | 369 | } */ |
---|
[24] | 370 | |
---|
[1] | 371 | if(strcmp(name,"account") == 0) { |
---|
| 372 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCOUNT,value); |
---|
| 373 | } |
---|
| 374 | else if(strcmp(name,"acctg-freq") == 0) { |
---|
| 375 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCTG_FREQ,value); |
---|
| 376 | } |
---|
| 377 | else if (strcmp(name,"comment") == 0) { |
---|
| 378 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_COMMENT,value); |
---|
| 379 | } |
---|
| 380 | else if (strcmp(name,"constraint") == 0) { |
---|
| 381 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONSTRAINT,value); |
---|
| 382 | } |
---|
| 383 | else if (strcmp(name,"contiguous") == 0) { |
---|
| 384 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONTIGUOUS,NULL); |
---|
| 385 | } |
---|
| 386 | else if(strcmp(name,"exclusive") == 0) { |
---|
| 387 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUSIVE,NULL); |
---|
| 388 | } |
---|
| 389 | else if (strcmp(name,"mem") == 0) { |
---|
| 390 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MEM,value); |
---|
| 391 | } |
---|
| 392 | else if (strcmp(name,"mem-per-cpu") == 0) { |
---|
| 393 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MEM_PER_CPU,value); |
---|
| 394 | } |
---|
| 395 | else if (strcmp(name,"mincpus") == 0) { |
---|
| 396 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MINCPUS,value); |
---|
| 397 | } |
---|
| 398 | else if (strcmp(name,"nodelist") == 0) { |
---|
| 399 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODELIST,value); |
---|
| 400 | } |
---|
| 401 | else if (strcmp(name,"nodes") == 0) { |
---|
| 402 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODES,value); |
---|
| 403 | } |
---|
| 404 | else if (strcmp(name,"ntasks-per-node") == 0) { |
---|
| 405 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS_PER_NODE,value); |
---|
| 406 | } |
---|
| 407 | else if (strcmp(name,"partition") == 0) { |
---|
| 408 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_PARTITION,value); |
---|
| 409 | } |
---|
| 410 | else if (strcmp(name,"qos") == 0) { |
---|
| 411 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_QOS,value); |
---|
| 412 | } |
---|
| 413 | else if (strcmp(name,"requeue") == 0) { |
---|
| 414 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_REQUEUE,NULL); |
---|
| 415 | } |
---|
| 416 | else if (strcmp(name,"reservation") == 0) { |
---|
| 417 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_RESERVATION,value); |
---|
| 418 | } |
---|
| 419 | else if (strcmp(name,"share") == 0) { |
---|
| 420 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_SHARE,NULL); |
---|
| 421 | } |
---|
[28] | 422 | else if(strcmp(name,"job_name") == 0) { |
---|
| 423 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_JOB_NAME,value); |
---|
| 424 | } |
---|
| 425 | else if(strcmp(name,"time_limit") == 0) { |
---|
| 426 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT,value); |
---|
| 427 | } |
---|
[22] | 428 | else if(strcmp(name,"time") == 0) { |
---|
[28] | 429 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT,value); |
---|
| 430 | } |
---|
| 431 | else if(strcmp(name,"ntasks") == 0) { |
---|
| 432 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS,value); |
---|
| 433 | } |
---|
| 434 | else if(strcmp(name,"gres") == 0) { |
---|
| 435 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_GRES,value); |
---|
| 436 | } |
---|
| 437 | else if(strcmp(name,"no-kill") == 0) { |
---|
| 438 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NO_KILL,NULL); |
---|
| 439 | } |
---|
| 440 | else if(strcmp(name,"licenses") == 0) { |
---|
| 441 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_LICENSES,value); |
---|
| 442 | } |
---|
| 443 | else if(strcmp(name,"mail-type") == 0) { |
---|
| 444 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MAIL_TYPE,value); |
---|
| 445 | } |
---|
| 446 | else if(strcmp(name,"no-requeue") == 0) { |
---|
| 447 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NO_REQUEUE,NULL); |
---|
| 448 | } |
---|
| 449 | else if(strcmp(name,"exclude") == 0) { |
---|
| 450 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUDE,value); |
---|
| 451 | } |
---|
| 452 | else { |
---|
[1] | 453 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE, |
---|
| 454 | "Invalid native specification: %s (Unsupported option: --%s)", |
---|
| 455 | add_attr, name); |
---|
| 456 | } |
---|
| 457 | |
---|
| 458 | } |
---|
| 459 | FINALLY |
---|
| 460 | { |
---|
| 461 | fsd_free(name); |
---|
| 462 | fsd_free(add_attr_copy); |
---|
| 463 | } |
---|
| 464 | END_TRY |
---|
| 465 | } |
---|
| 466 | |
---|
| 467 | void |
---|
| 468 | slurmdrmaa_parse_native(job_desc_msg_t *job_desc, const char * value) |
---|
| 469 | { |
---|
| 470 | char *arg = NULL; |
---|
| 471 | char * volatile native_specification = fsd_strdup(value); |
---|
| 472 | char * volatile native_spec_copy = fsd_strdup(native_specification); |
---|
| 473 | char * ctxt = NULL; |
---|
| 474 | int opt = 0; |
---|
| 475 | |
---|
| 476 | fsd_log_enter(( "" )); |
---|
| 477 | TRY |
---|
| 478 | { |
---|
| 479 | for (arg = strtok_r(native_spec_copy, " \t", &ctxt); arg; arg = strtok_r(NULL, " \t",&ctxt) ) { |
---|
| 480 | if (!opt) { |
---|
| 481 | if ( (arg[0] != '-') || ((strlen(arg) != 2) && (strlen(arg) > 2) && arg[2] != ' ' && arg[1] !='-' ) ) { |
---|
| 482 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE, |
---|
| 483 | "Invalid native specification: %s", |
---|
| 484 | native_specification); |
---|
| 485 | } |
---|
| 486 | if(arg[1] == '-') { |
---|
| 487 | slurmdrmaa_parse_additional_attr(job_desc, arg+2); |
---|
| 488 | } |
---|
| 489 | else { |
---|
| 490 | opt = arg[1]; |
---|
| 491 | } |
---|
| 492 | } else { |
---|
[28] | 493 | switch (opt) { |
---|
[1] | 494 | case 'A' : |
---|
| 495 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCOUNT, arg); |
---|
| 496 | break; |
---|
| 497 | case 'C' : |
---|
| 498 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONSTRAINT, arg); |
---|
| 499 | break; |
---|
| 500 | case 'N' : |
---|
| 501 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODES, arg); |
---|
| 502 | break; |
---|
| 503 | case 'p' : |
---|
| 504 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_PARTITION, arg); |
---|
| 505 | break; |
---|
| 506 | case 's' : |
---|
| 507 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_SHARE, NULL); |
---|
[28] | 508 | break; |
---|
[1] | 509 | case 'w' : |
---|
| 510 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODELIST, arg); |
---|
[18] | 511 | break; |
---|
| 512 | case 'J' : |
---|
| 513 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_JOB_NAME, arg); |
---|
[28] | 514 | break; |
---|
[24] | 515 | case 't' : |
---|
| 516 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT, arg); |
---|
| 517 | break; |
---|
| 518 | case 'n' : |
---|
| 519 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS, arg); |
---|
| 520 | break; |
---|
[28] | 521 | case 'x' : |
---|
| 522 | slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUDE, arg); |
---|
| 523 | break; |
---|
| 524 | default : |
---|
[1] | 525 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE, |
---|
| 526 | "Invalid native specification: %s (Unsupported option: -%c)", |
---|
| 527 | native_specification, opt); |
---|
| 528 | } |
---|
| 529 | if (arg[0] == '-' && strlen(arg) == 2) { /* attribute without arg */ |
---|
| 530 | opt = arg[1]; |
---|
| 531 | } |
---|
| 532 | else { |
---|
| 533 | opt = 0; |
---|
| 534 | } |
---|
| 535 | } |
---|
| 536 | } |
---|
| 537 | |
---|
| 538 | if(strlen(native_spec_copy) == 2 && native_spec_copy[0] == '-' && native_spec_copy[1] == 's') |
---|
| 539 | { |
---|
| 540 | slurmdrmaa_add_attribute(job_desc, SLURM_NATIVE_SHARE, NULL); |
---|
| 541 | opt = 0; |
---|
| 542 | } |
---|
| 543 | |
---|
| 544 | if (opt) |
---|
| 545 | fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE, |
---|
| 546 | "Invalid native specification: %s", |
---|
| 547 | native_specification); |
---|
| 548 | |
---|
| 549 | } |
---|
| 550 | FINALLY |
---|
| 551 | { |
---|
| 552 | fsd_free(native_spec_copy); |
---|
| 553 | fsd_free(native_specification); |
---|
| 554 | } |
---|
| 555 | END_TRY |
---|
| 556 | |
---|
| 557 | fsd_log_return(( "" )); |
---|
| 558 | } |
---|
| 559 | |
---|