openmp/runtime/test/affinity/format/api.c

   1 // RUN: %libomp-compile-and-run
   2 // RUN: %libomp-run | %python %S/check.py -c 'CHECK' %s
   3
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <omp.h>
   8
   9 #define XSTR(x) #x
  10 #define STR(x) XSTR(x)
  11
  12 #define streqls(s1, s2) (!strcmp(s1, s2))
  13
  14 #define check(condition)                                                       \
  15   if (!(condition)) {                                                          \
  16     fprintf(stderr, "error: %s: %d: " STR(condition) "\n", __FILE__,           \
  17             __LINE__);                                                         \
  18     exit(1);                                                                   \
  19   }
  20
  21 #define BUFFER_SIZE 1024
  22
  23 int main(int argc, char** argv) {
  24   char buf[BUFFER_SIZE];
  25   size_t needed;
  26
  27   omp_set_affinity_format("0123456789");
  28
  29   needed = omp_get_affinity_format(buf, BUFFER_SIZE);
  30   check(streqls(buf, "0123456789"));
  31   check(needed == 10)
  32
  33   // Check that it is truncated properly
  34   omp_get_affinity_format(buf, 5);
  35   check(streqls(buf, "0123"));
  36
  37   #pragma omp parallel
  38   {
  39     char my_buf[512];
  40     size_t needed = omp_capture_affinity(my_buf, 512, NULL);
  41     check(streqls(my_buf, "0123456789"));
  42     check(needed == 10);
  43     // Check that it is truncated properly
  44     omp_capture_affinity(my_buf, 5, NULL);
  45     check(streqls(my_buf, "0123"));
  46   }
  47
  48   #pragma omp parallel num_threads(4)
  49   {
  50     omp_display_affinity(NULL);
  51   }
  52
  53   return 0;
  54 }
  55
  56 // CHECK: num_threads=4 0123456789