Tests: Add a test that check the binding of workers when STARPU_WORKERS_CPUID is set

tests/Makefile.am

@@ -220,6 +220,7 @@ noinst_PROGRAMS =				\
 	errorcheck/starpu_init_noworker		\
 	errorcheck/invalid_blocking_calls	\
 	errorcheck/invalid_tasks		\
+	errorcheck/workers_cpuid		\
 	helper/cublas_init			\
 	helper/starpu_data_cpy			\
 	helper/pinned_memory			\

tests/errorcheck/workers_cpuid.c

@@ -0,0 +1,195 @@
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
+ *
+ * Copyright (C) 2010-2012  Université de Bordeaux
+ * Copyright (C) 2010, 2011, 2012, 2013  CNRS
+ *
+ * StarPU is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 of the License, or (at
+ * your option) any later version.
+ *
+ * StarPU is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * See the GNU Lesser General Public License in COPYING.LGPL for more details.
+ */
+
+#include <starpu.h>
+#include "../helper.h"
+#include <common/config.h>
+#include <hwloc.h>
+
+
+#if !defined(STARPU_HAVE_UNSETENV) || !defined(STARPU_USE_CPU)
+#warning unsetenv is not defined or no cpu are available. Skipping test
+int main(int argc, char **argv)
+{
+	return STARPU_TEST_SKIPPED;
+}
+#else
+
+#define NB_TESTS 5
+
+int nhwpus;
+long workers_cpuid[STARPU_NMAXWORKERS];
+int workers_id[STARPU_NMAXWORKERS];
+
+
+
+static int check_workers_mapping(long *cpuid, int *workerids, int nb_workers)
+{
+	int i;
+	for (i=0; i<nb_workers; i++)
+	{
+		int workerid = workerids[i];
+		long bindid = starpu_worker_get_bindid(workerid);
+		if ( bindid != cpuid[i])
+		{
+			fprintf(stderr, "Worker %d (%s) is on cpu %ld rather than on %ld\n",
+			       workerid, starpu_worker_get_type_as_string(starpu_worker_get_type(workerid)),
+			       bindid, cpuid[i]);
+			return 0;
+		}
+	}
+	return 1;
+}
+
+
+static void copy_cpuid_array(long *dst, long *src, unsigned n)
+{
+	int i;
+
+	memcpy(dst, src, n * sizeof(long));
+	for (i=n; i<STARPU_NMAXWORKERS; i++)
+		dst[i] = src[i%n];
+}
+
+static char *array_to_str(long *array, int n)
+{
+	int i;
+	char *str = malloc(n * 3 * sizeof(long));
+	char *ptr = str;
+
+	for (i=0; i<n; i++)
+	{
+		int nchar;
+		nchar = sprintf(ptr, "%ld ", array[i]);
+		ptr += nchar;
+	}
+		
+	return str;
+}
+
+static int test_combination(long *combination, unsigned n)
+{
+	int type, ret, device_workers;
+
+	copy_cpuid_array(workers_cpuid, combination, n);
+
+	setenv("STARPU_WORKERS_CPUID", array_to_str(workers_cpuid, n), 1);
+
+	ret = starpu_init(NULL);
+	if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
+	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
+
+	device_workers = 0;
+	
+	/* Check for all devices */
+	for (type=STARPU_CUDA_WORKER; type<STARPU_NARCH; type++)
+	{
+		int nb_workers;
+		nb_workers = starpu_worker_get_ids_by_type(type, workers_id, starpu_worker_get_count());
+		device_workers += nb_workers ;
+		if (!check_workers_mapping(workers_cpuid, workers_id, nb_workers))
+			return -1;
+	}
+
+	/* Check for all cpus */
+	{
+		int nb_workers;
+		nb_workers = starpu_worker_get_ids_by_type(STARPU_CPU_WORKER, workers_id, starpu_worker_get_count());
+		if (!check_workers_mapping(workers_cpuid + device_workers, workers_id, nb_workers))
+			return -1;
+	}
+	
+	starpu_shutdown();
+	return 1;
+}
+
+
+static long * generate_arrangement(int arr_size, long *set, int set_size)
+{
+	int i, j;
+	long tmp;
+
+	STARPU_ASSERT(arr_size <= set_size);
+	
+	srandom(time(0));
+
+	for (i=0; i<arr_size; i++)
+	{
+		/* Pick a random value in the set */
+		j = random() % (set_size - i);
+
+		/* Switch the value picked up with the beginning value of set */
+		tmp = set[i+j];
+		set[i] = set[i+j];
+		set[i+j] = tmp;
+	}
+	
+	return set;
+}
+
+
+static void init_array(long *a, int n)
+{
+	int i;
+
+	for(i=0; i<n; i++)
+		a[i] = i;
+}
+
+
+int main(int argc, char **argv)
+{
+	int ret, i, j;
+	long *cpuids;
+	hwloc_topology_t topology;
+
+	hwloc_topology_init(&topology);
+	hwloc_topology_load(topology);
+	nhwpus = hwloc_get_nbobjs_by_type(topology, HWLOC_OBJ_PU);
+
+	unsetenv("STARPU_NCUDA");
+	unsetenv("STARPU_NOPENCL");
+	unsetenv("STARPU_NCPUS");
+
+	for (i=0; i<STARPU_NMAXWORKERS; i++)
+		workers_id[i] = -1;
+		
+	cpuids = malloc(nhwpus * sizeof(long));
+
+	/* Evaluate several random values of STARPU_WORKERS_CPUID 
+	 * and check mapping for each one
+	 */
+	for (i=1; i<=nhwpus; i++)
+	{
+		for (j=0; j<NB_TESTS; j++)
+		{
+			init_array(cpuids, nhwpus);
+			generate_arrangement(i, cpuids, nhwpus);
+			test_combination(cpuids, i);
+		}
+	}
+			
+	free(cpuids);
+
+	hwloc_topology_destroy(topology);
+
+	return EXIT_SUCCESS;
+}
+#endif
+
+
+