starpu_init_noworker.c 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009, 2010, 2012-2014 Université de Bordeaux
  4. * Copyright (C) 2010, 2011, 2012, 2013 CNRS
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. #include <stdio.h>
  18. #include <unistd.h>
  19. #include <errno.h>
  20. #include <starpu.h>
  21. #include <stdlib.h>
  22. #include "../helper.h"
  23. #if !defined(STARPU_HAVE_UNSETENV)
  24. #warning unsetenv is not defined. Skipping test
  25. int main(int argc, char **argv)
  26. {
  27. return STARPU_TEST_SKIPPED;
  28. }
  29. #else
  30. static void unset_env_variables(void)
  31. {
  32. (void) unsetenv("STARPU_NCPUS");
  33. (void) unsetenv("STARPU_NCPU");
  34. (void) unsetenv("STARPU_NCUDA");
  35. (void) unsetenv("STARPU_NOPENCL");
  36. (void) unsetenv("STARPU_NMIC");
  37. (void) unsetenv("STARPU_NSCC");
  38. }
  39. int main(int argc, char **argv)
  40. {
  41. int ret;
  42. unset_env_variables();
  43. #ifdef STARPU_USE_MIC
  44. #ifdef STARPU_DEVEL
  45. #warning nmic = 0 make initialization hang and leaves a process behind...
  46. #endif
  47. return STARPU_TEST_SKIPPED;
  48. #endif
  49. /* We try to initialize StarPU without any worker */
  50. struct starpu_conf conf;
  51. starpu_conf_init(&conf);
  52. conf.ncpus = 0;
  53. conf.ncuda = 0;
  54. conf.nopencl = 0;
  55. conf.nmic = 0;
  56. conf.nscc = 0;
  57. /* starpu_init should return -ENODEV */
  58. ret = starpu_initialize(&conf, &argc, &argv);
  59. if (ret == -ENODEV)
  60. return EXIT_SUCCESS;
  61. else
  62. {
  63. unsigned ncpu = starpu_cpu_worker_get_count();
  64. unsigned ncuda = starpu_cuda_worker_get_count();
  65. unsigned nopencl = starpu_opencl_worker_get_count();
  66. FPRINTF(stderr, "StarPU has found :\n");
  67. FPRINTF(stderr, "\t%u CPU cores\n", ncpu);
  68. FPRINTF(stderr, "\t%u CUDA devices\n", ncuda);
  69. FPRINTF(stderr, "\t%u OpenCL devices\n", nopencl);
  70. return EXIT_FAILURE;
  71. }
  72. }
  73. #endif