Context Navigation

← Previous Revision
Next Revision →
Normal
Revision Log

util.c @ 43

Revision 43, 16.5 KB checked in by pkopta, 10 years ago (diff)
dependency option by Rasmus Rohde <rohde@…>
Property svn:executable set to ``* Property svn:keywords set to `Id`

Rev	Line
[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;
	96	else if (strcasecmp(mail_type_str, "REQUEUE") == 0)
	97	rc = MAIL_JOB_REQUEUE;
	98	else if (strcasecmp(mail_type_str, "ALL") == 0)
	99	rc = MAIL_JOB_BEGIN \| MAIL_JOB_END \| MAIL_JOB_FAIL \| MAIL_JOB_REQUEUE;
[29]	100	else {
	101	fsd_log_error(("Unknown mail type: %s", mail_type_str));
	102	}
[28]	103
	104	return rc;
	105	}
	106
	107
[1]	108	enum slurm_native {
	109	SLURM_NATIVE_ACCOUNT,
	110	SLURM_NATIVE_ACCTG_FREQ,
	111	SLURM_NATIVE_COMMENT,
	112	SLURM_NATIVE_CONSTRAINT,
	113	SLURM_NATIVE_CONTIGUOUS,
[35]	114	SLURM_NATIVE_CPUS_PER_TASK,
[1]	115	SLURM_NATIVE_EXCLUSIVE,
	116	SLURM_NATIVE_MEM,
	117	SLURM_NATIVE_MEM_PER_CPU,
	118	SLURM_NATIVE_MINCPUS,
	119	SLURM_NATIVE_NODELIST,
	120	SLURM_NATIVE_NODES,
	121	SLURM_NATIVE_NTASKS_PER_NODE,
	122	SLURM_NATIVE_PARTITION,
	123	SLURM_NATIVE_QOS,
	124	SLURM_NATIVE_REQUEUE,
	125	SLURM_NATIVE_RESERVATION,
[18]	126	SLURM_NATIVE_SHARE,
[19]	127	SLURM_NATIVE_JOB_NAME,
[24]	128	SLURM_NATIVE_TIME_LIMIT,
[28]	129	SLURM_NATIVE_NTASKS,
	130	SLURM_NATIVE_GRES,
	131	SLURM_NATIVE_NO_KILL,
	132	SLURM_NATIVE_LICENSES,
	133	SLURM_NATIVE_MAIL_TYPE,
	134	SLURM_NATIVE_NO_REQUEUE,
[30]	135	SLURM_NATIVE_EXCLUDE,
[43]	136	SLURM_NATIVE_TMP,
	137	SLURM_NATIVE_DEPENDENCY
[1]	138	};
	139
	140	void
	141	slurmdrmaa_init_job_desc(job_desc_msg_t *job_desc)
	142	{
	143	memset(job_desc, 0, sizeof(job_desc_msg_t));
	144	}
	145
	146	void
	147	slurmdrmaa_free_job_desc(job_desc_msg_t *job_desc)
	148	{
	149	unsigned i = 0;
	150	fsd_log_enter(( "" ));
	151	fsd_free(job_desc->account);
	152	fsd_free(job_desc->comment);
	153
	154	for(i = 0;i<job_desc->env_size;i++)
	155	{
	156	fsd_free(job_desc->environment[i]);
	157	}
	158	fsd_free(job_desc->environment);
	159	fsd_free(job_desc->features);
	160	fsd_free(job_desc->name);
	161	fsd_free(job_desc->mail_user);
	162	fsd_free(job_desc->partition);
	163	fsd_free(job_desc->qos);
	164	fsd_free(job_desc->script);
	165	fsd_free(job_desc->std_in);
	166	fsd_free(job_desc->std_out);
	167	fsd_free(job_desc->std_err);
	168	fsd_free(job_desc->work_dir);
[29]	169	fsd_free(job_desc->exc_nodes);
[28]	170	fsd_free(job_desc->gres);
[38]	171	fsd_free(job_desc->array_inx);
[1]	172
	173	fsd_log_return(( "" ));
	174	}
	175
	176	void
	177	slurmdrmaa_add_attribute(job_desc_msg_t job_desc, unsigned attr, const char value)
	178	{
	179	char * ptr = NULL;
	180	char * rest = NULL;
	181	char * token = NULL;
	182
	183	switch(attr)
	184	{
	185	case SLURM_NATIVE_ACCOUNT:
	186	fsd_free(job_desc->account);
	187	fsd_log_debug(("# account = %s",value));
	188	job_desc->account = fsd_strdup(value);
	189	break;
	190	case SLURM_NATIVE_ACCTG_FREQ:
	191	fsd_log_debug(("# acctg_freq = %s",value));
	192	job_desc->acctg_freq = fsd_atoi(value);
	193	break;
	194	case SLURM_NATIVE_COMMENT:
	195	fsd_free(job_desc->comment);
	196	fsd_log_debug(("# comment = %s",value));
	197	job_desc->comment = fsd_strdup(value);
	198	break;
	199	case SLURM_NATIVE_CONSTRAINT:
	200	fsd_free(job_desc->features);
	201	fsd_log_debug(("# constraints -> features = %s",value));
	202	job_desc->features = fsd_strdup(value);
	203	break;
	204	case SLURM_NATIVE_CONTIGUOUS:
	205	fsd_log_debug(( "# contiguous = 1"));
	206	job_desc->contiguous = 1;
	207	break;
[35]	208	case SLURM_NATIVE_CPUS_PER_TASK:
	209	fsd_log_debug(( "# cpus_per_task = %s", value));
	210	job_desc->cpus_per_task = fsd_atoi(value);
	211	break;
[1]	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(job_desc->pn_min_memory == NO_VAL \|\| fsd_atoi(value) > (int)job_desc->pn_min_memory) {
	218	fsd_log_debug(("# pn_min_memory = %s",value));
	219	job_desc->pn_min_memory = fsd_atoi(value);
	220	}
	221	else {
[1]	222	fsd_log_debug(("mem value defined lower or equal to mem-per-cpu or value defined before"));
	223	}
	224	break;
	225	case SLURM_NATIVE_MEM_PER_CPU:
[13]	226	if(job_desc->pn_min_memory == NO_VAL \|\| fsd_atoi(value) > (int)job_desc->pn_min_memory) {
[31]	227	fsd_log_debug(("# pn_min_memory (MEM_PER_CPU) = %s",value));
	228	job_desc->pn_min_memory = fsd_atoi(value) \| MEM_PER_CPU;
[13]	229	}
	230	else {
[1]	231	fsd_log_debug(("mem-per-cpu value defined lower or equal to mem or value defined before"));
	232	}
	233	break;
	234	case SLURM_NATIVE_MINCPUS:
[35]	235	fsd_log_debug(("# pn_min_cpus = %s",value));
	236	job_desc->pn_min_cpus = fsd_atoi(value);
[1]	237	break;
	238	case SLURM_NATIVE_NODELIST:
	239	fsd_free(job_desc->req_nodes);
	240	fsd_log_debug(("# node-list - > req_nodes = %s",value));
	241	job_desc->req_nodes = fsd_strdup(value);
	242	break;
	243	case SLURM_NATIVE_NODES:
	244	ptr = strdup(value);
[24]	245
[1]	246	if((token = strtok_r(ptr,"=",&rest)) == NULL){
	247	fsd_log_error(("strtok_r returned NULL"));
	248	}
	249
	250	fsd_log_debug(("nodes: %s ->",value));
	251	fsd_log_debug(("# min_nodes = %s",token));
	252	job_desc->min_nodes = fsd_atoi(token);
	253
	254	if(strcmp(rest,"") !=0 ) {
	255	fsd_log_debug(("# max_nodes = %s",rest));
	256	job_desc->max_nodes = fsd_atoi(rest);
	257	}
	258	fsd_free(ptr);
	259	break;
	260	case SLURM_NATIVE_NTASKS_PER_NODE:
	261	fsd_log_debug(("# ntasks_per_node = %s",value));
	262	job_desc->ntasks_per_node = fsd_atoi(value);
	263	break;
	264	case SLURM_NATIVE_PARTITION:
	265	fsd_free(job_desc->partition);
	266	fsd_log_debug(("# partition = %s",value));
	267	job_desc->partition = fsd_strdup(value);
	268	break;
	269	case SLURM_NATIVE_QOS:
	270	fsd_free(job_desc->qos);
	271	fsd_log_debug(("# qos = %s",value));
	272	job_desc->qos = fsd_strdup(value);
	273	break;
	274	case SLURM_NATIVE_REQUEUE:
	275	fsd_log_debug(( "# requeue = 1"));
	276	job_desc->requeue = 1;
	277	break;
	278	case SLURM_NATIVE_RESERVATION:
	279	fsd_log_debug(("# reservation = %s",value));
	280	job_desc->reservation = fsd_strdup(value);
	281	break;
	282	case SLURM_NATIVE_SHARE:
	283	fsd_log_debug(("# shared = 1"));
	284	job_desc->shared = 1;
	285	break;
[18]	286	case SLURM_NATIVE_JOB_NAME:
[24]	287	fsd_log_debug(("# job_name = %s",job_desc->name));
	288	job_desc->name = fsd_strdup(value);
[18]	289	break;
[24]	290	case SLURM_NATIVE_NTASKS:
	291	fsd_log_debug(("# ntasks = %s",value));
	292	job_desc->num_tasks = fsd_atoi(value);
	293	break;
	294	case SLURM_NATIVE_TIME_LIMIT:
	295	fsd_log_debug(("# time_limit = %s",value));
	296	job_desc->time_limit = slurmdrmaa_datetime_parse(value);
	297	break;
[28]	298	case SLURM_NATIVE_GRES:
	299	fsd_log_debug(("# gres = %s",value));
	300	job_desc->gres = fsd_strdup(value);
	301	break;
	302	case SLURM_NATIVE_NO_KILL:
	303	fsd_log_debug(("# no_kill = 1"));
	304	job_desc->kill_on_node_fail = 0;
	305	break;
	306	case SLURM_NATIVE_LICENSES:
	307	fsd_log_debug(("# licenses = %s", value));
	308	job_desc->licenses = fsd_strdup(value);
	309	break;
	310	case SLURM_NATIVE_MAIL_TYPE:
	311	fsd_log_debug(("# mail_type = %s", value));
	312	job_desc->mail_type = slurmdrmaa_mail_type_parse(value);
	313	break;
	314	case SLURM_NATIVE_NO_REQUEUE:
	315	fsd_log_debug(("# requeue = 0"));
	316	job_desc->requeue = 0;
	317	break;
	318	case SLURM_NATIVE_EXCLUDE:
	319	fsd_log_debug(("# exclude = %s", value));
	320	job_desc->exc_nodes = fsd_strdup(value);
	321	break;
[30]	322	case SLURM_NATIVE_TMP:
	323	fsd_log_debug(("# tmp = %s", value));
[43]	324	job_desc->pn_min_tmp_disk = fsd_atoi(value);
[30]	325	break;
[43]	326	case SLURM_NATIVE_DEPENDENCY:
	327	fsd_log_debug(("# dependency = %s", value));
	328	job_desc->dependency = fsd_strdup(value);
	329	break;
[1]	330	default:
	331	fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE,"Invalid attribute");
	332	}
	333	}
	334
	335	void
	336	slurmdrmaa_parse_additional_attr(job_desc_msg_t job_desc,const char add_attr)
	337	{
	338	char * volatile name = NULL;
	339	char *value = NULL;
	340	char *ctxt = NULL;
	341	char * volatile add_attr_copy = fsd_strdup(add_attr);
	342
[35]	343	fsd_log_enter(( "" ));
	344
[1]	345	TRY
	346	{
	347	name = fsd_strdup(strtok_r(add_attr_copy, "=", &ctxt));
	348	value = strtok_r(NULL, "=", &ctxt);
[28]	349	/*
	350	* TODO: move it to slurmdrmaa_add_attribute
	351	if (value == NULL) {
[24]	352	fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE,
	353	"Invalid native specification: %s Missing '='.", add_attr_copy);
[28]	354	} */
[24]	355
[1]	356	if(strcmp(name,"account") == 0) {
	357	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCOUNT,value);
	358	}
	359	else if(strcmp(name,"acctg-freq") == 0) {
	360	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCTG_FREQ,value);
	361	}
	362	else if (strcmp(name,"comment") == 0) {
	363	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_COMMENT,value);
	364	}
	365	else if (strcmp(name,"constraint") == 0) {
	366	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONSTRAINT,value);
	367	}
	368	else if (strcmp(name,"contiguous") == 0) {
	369	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONTIGUOUS,NULL);
	370	}
[35]	371	else if (strcmp(name,"cpus-per-task") == 0) {
	372	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CPUS_PER_TASK,value);
	373	}
[1]	374	else if(strcmp(name,"exclusive") == 0) {
	375	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUSIVE,NULL);
	376	}
	377	else if (strcmp(name,"mem") == 0) {
	378	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MEM,value);
	379	}
	380	else if (strcmp(name,"mem-per-cpu") == 0) {
	381	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MEM_PER_CPU,value);
	382	}
	383	else if (strcmp(name,"mincpus") == 0) {
	384	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MINCPUS,value);
	385	}
	386	else if (strcmp(name,"nodelist") == 0) {
	387	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODELIST,value);
	388	}
	389	else if (strcmp(name,"nodes") == 0) {
	390	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODES,value);
	391	}
	392	else if (strcmp(name,"ntasks-per-node") == 0) {
	393	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS_PER_NODE,value);
	394	}
	395	else if (strcmp(name,"partition") == 0) {
	396	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_PARTITION,value);
	397	}
	398	else if (strcmp(name,"qos") == 0) {
	399	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_QOS,value);
	400	}
	401	else if (strcmp(name,"requeue") == 0) {
	402	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_REQUEUE,NULL);
	403	}
	404	else if (strcmp(name,"reservation") == 0) {
	405	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_RESERVATION,value);
	406	}
	407	else if (strcmp(name,"share") == 0) {
	408	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_SHARE,NULL);
	409	}
[28]	410	else if(strcmp(name,"job_name") == 0) {
	411	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_JOB_NAME,value);
	412	}
	413	else if(strcmp(name,"time_limit") == 0) {
	414	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT,value);
	415	}
[22]	416	else if(strcmp(name,"time") == 0) {
[28]	417	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT,value);
	418	}
	419	else if(strcmp(name,"ntasks") == 0) {
	420	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS,value);
	421	}
	422	else if(strcmp(name,"gres") == 0) {
	423	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_GRES,value);
	424	}
	425	else if(strcmp(name,"no-kill") == 0) {
	426	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NO_KILL,NULL);
	427	}
	428	else if(strcmp(name,"licenses") == 0) {
	429	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_LICENSES,value);
	430	}
	431	else if(strcmp(name,"mail-type") == 0) {
	432	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_MAIL_TYPE,value);
	433	}
	434	else if(strcmp(name,"no-requeue") == 0) {
	435	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NO_REQUEUE,NULL);
	436	}
	437	else if(strcmp(name,"exclude") == 0) {
	438	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUDE,value);
	439	}
[30]	440	else if(strcmp(name,"tmp") == 0) {
	441	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TMP,value);
[43]	442	}
	443	else if(strcmp(name,"dependency") == 0) {
	444	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_DEPENDENCY,value);
[30]	445	} else {
[1]	446	fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE,
	447	"Invalid native specification: %s (Unsupported option: --%s)",
	448	add_attr, name);
	449	}
	450
	451	}
	452	FINALLY
	453	{
	454	fsd_free(name);
	455	fsd_free(add_attr_copy);
	456	}
	457	END_TRY
[35]	458	fsd_log_return(( "" ));
[1]	459	}
	460
	461	void
	462	slurmdrmaa_parse_native(job_desc_msg_t job_desc, const char value)
	463	{
	464	char *arg = NULL;
	465	char * volatile native_specification = fsd_strdup(value);
	466	char * volatile native_spec_copy = fsd_strdup(native_specification);
	467	char * ctxt = NULL;
	468	int opt = 0;
	469
	470	fsd_log_enter(( "" ));
	471	TRY
	472	{
	473	for (arg = strtok_r(native_spec_copy, " \t", &ctxt); arg; arg = strtok_r(NULL, " \t",&ctxt) ) {
	474	if (!opt) {
	475	if ( (arg[0] != '-') \|\| ((strlen(arg) != 2) && (strlen(arg) > 2) && arg[2] != ' ' && arg[1] !='-' ) ) {
	476	fsd_exc_raise_fmt(FSD_DRMAA_ERRNO_INVALID_ATTRIBUTE_VALUE,
	477	"Invalid native specification: %s",
	478	native_specification);
	479	}
	480	if(arg[1] == '-') {
	481	slurmdrmaa_parse_additional_attr(job_desc, arg+2);
	482	}
	483	else {
	484	opt = arg[1];
	485	}
	486	} else {
[28]	487	switch (opt) {
[1]	488	case 'A' :
	489	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_ACCOUNT, arg);
	490	break;
	491	case 'C' :
	492	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CONSTRAINT, arg);
	493	break;
[35]	494	case 'c' :
	495	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_CPUS_PER_TASK, arg);
	496	break;
[1]	497	case 'N' :
	498	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODES, arg);
	499	break;
	500	case 'p' :
	501	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_PARTITION, arg);
	502	break;
	503	case 's' :
	504	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_SHARE, NULL);
[28]	505	break;
[1]	506	case 'w' :
	507	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NODELIST, arg);
[18]	508	break;
	509	case 'J' :
	510	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_JOB_NAME, arg);
[28]	511	break;
[24]	512	case 't' :
	513	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_TIME_LIMIT, arg);
	514	break;
	515	case 'n' :
	516	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_NTASKS, arg);
	517	break;
[28]	518	case 'x' :
	519	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_EXCLUDE, arg);
	520	break;
[33]	521	case 'L' :
	522	slurmdrmaa_add_attribute(job_desc,SLURM_NATIVE_LICENSES, arg);
	523	break;
[28]	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	{
[35]	552	fsd_log_debug(( "finalizing job constraints" ));
	553	if( job_desc->cpus_per_task > 0 ) {
	554	job_desc->min_cpus = job_desc->num_tasks * job_desc->cpus_per_task ;
	555	fsd_log_debug((
	556	"set min_cpus to ntasks*cpus_per_task: %d",
	557	job_desc->min_cpus
	558	));
	559	} else {
	560	job_desc->min_cpus = job_desc->num_tasks ;
	561	fsd_log_debug((
	562	"set min_cpus to ntasks: %d",
	563	job_desc->min_cpus
	564	));
	565	}
[1]	566	fsd_free(native_spec_copy);
	567	fsd_free(native_specification);
	568	}
	569	END_TRY
	570
	571	fsd_log_return(( "" ));
	572	}
	573

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: branches/slurm-drmaa-1.2.0/slurm_drmaa/util.c @ 43

Download in other formats: