starpu_init_noworker.c 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009, 2010, 2012 Université de Bordeaux 1
  4. * Copyright (C) 2010, 2011, 2012, 2013 Centre National de la Recherche Scientifique
  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_NCUDA");
  34. (void) unsetenv("STARPU_NOPENCL");
  35. (void) unsetenv("STARPU_NMIC");
  36. (void) unsetenv("STARPU_NSCC");
  37. }
  38. int main(int argc, char **argv)
  39. {
  40. int ret;
  41. unset_env_variables();
  42. /* We try to initialize StarPU without any worker */
  43. struct starpu_conf conf;
  44. starpu_conf_init(&conf);
  45. conf.ncpus = 0;
  46. conf.ncuda = 0;
  47. conf.nopencl = 0;
  48. conf.nmic = 0;
  49. conf.nscc = 0;
  50. /* starpu_init should return -ENODEV */
  51. ret = starpu_initialize(&conf, &argc, &argv);
  52. if (ret == -ENODEV)
  53. return EXIT_SUCCESS;
  54. else
  55. {
  56. unsigned ncpu = starpu_cpu_worker_get_count();
  57. unsigned ncuda = starpu_cuda_worker_get_count();
  58. unsigned nopencl = starpu_opencl_worker_get_count();
  59. FPRINTF(stderr, "StarPU has found :\n");
  60. FPRINTF(stderr, "\t%u CPU cores\n", ncpu);
  61. FPRINTF(stderr, "\t%u CUDA devices\n", ncuda);
  62. FPRINTF(stderr, "\t%u OpenCL devices\n", nopencl);
  63. return EXIT_FAILURE;
  64. }
  65. }
  66. #endif