Bläddra i källkod

socl/src/init.c: partly revert revision 6585
- override user environment to force STARPU_NCUDA to 0
- override user environment to force STARPU_NCPUS to 1 if it is set to 0 as socl needs at least 1 CPU device

Nathalie Furmento 13 år sedan
förälder
incheckning
2c5f317f29
1 ändrade filer med 5 tillägg och 0 borttagningar
  1. 5 0
      socl/src/init.c

+ 5 - 0
socl/src/init.c

@@ -26,8 +26,13 @@ int _starpu_init_failed;
 __attribute__((constructor)) static void socl_init() {
 
   struct starpu_conf conf;
+  const char *env;
+
   starpu_conf_init(&conf);
   conf.ncuda = 0;
+  putenv("STARPU_NCUDA=0");
+  if ((env = getenv("STARPU_NCPUS")) && atoi(env) == 0)
+	  putenv("STARPU_NCPUS=1");
 
   mem_object_init();