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

   1 // RUN: %libomp-compile-and-run
   2
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <omp.h>
   7
   8 #define XSTR(x) #x
   9 #define STR(x) XSTR(x)
  10
  11 #define streqls(s1, s2) (!strcmp(s1, s2))
  12
  13 #define check(condition)                                                       \
  14   if (!(condition)) {                                                          \
  15     fprintf(stderr, "error: %s: %d: " STR(condition) "\n", __FILE__,           \
  16             __LINE__);                                                         \
  17     exit(1);                                                                   \
  18   }
  19
  20 #define BUFFER_SIZE 1024
  21
  22 char buf[BUFFER_SIZE];
  23 #pragma omp threadprivate(buf)
  24
  25 char* get_string(size_t check_needed) {
  26   size_t needed = omp_capture_affinity(buf, BUFFER_SIZE, NULL);
  27   //printf("buf = %s\n", buf);
  28   check(needed < BUFFER_SIZE);
  29   if (check_needed != 0) {
  30     check(needed == check_needed);
  31   }
  32   return buf;
  33 }
  34
  35 void check_thread_num_padded_rjustified() {
  36   int i;
  37   const char* formats[2] = {"%0.8{thread_num}", "%0.8n"};
  38   for (i = 0; i < sizeof(formats)/sizeof(formats[0]); ++i) {
  39     omp_set_affinity_format(formats[i]);
  40     #pragma omp parallel num_threads(8)
  41     {
  42       int j;
  43       int tid = omp_get_thread_num();
  44       char ctid = '0' + (char)tid;
  45       char* s = get_string(8);
  46       for (j = 0; j < 7; ++j) {
  47         check(s[j] == '0');
  48       }
  49       check(s[j] == ctid);
  50     }
  51   }
  52 }
  53
  54 void check_thread_num_rjustified() {
  55   int i;
  56   const char* formats[2] = {"%.12{thread_num}", "%.12n"};
  57   for (i = 0; i < sizeof(formats)/sizeof(formats[0]); ++i) {
  58     omp_set_affinity_format(formats[i]);
  59     #pragma omp parallel num_threads(8)
  60     {
  61       int j;
  62       int tid = omp_get_thread_num();
  63       char ctid = '0' + (char)tid;
  64       char* s = get_string(12);
  65       for (j = 0; j < 11; ++j) {
  66         check(s[j] == ' ');
  67       }
  68       check(s[j] == ctid);
  69     }
  70   }
  71 }
  72
  73 void check_thread_num_ljustified() {
  74   int i;
  75   const char* formats[2] = {"%5{thread_num}", "%5n"};
  76   for (i = 0; i < sizeof(formats)/sizeof(formats[0]); ++i) {
  77     omp_set_affinity_format(formats[i]);
  78     #pragma omp parallel num_threads(8)
  79     {
  80       int j;
  81       int tid = omp_get_thread_num();
  82       char ctid = '0' + (char)tid;
  83       char* s = get_string(5);
  84       check(s[0] == ctid);
  85       for (j = 1; j < 5; ++j) {
  86         check(s[j] == ' ');
  87       }
  88     }
  89   }
  90 }
  91
  92 void check_thread_num_padded_ljustified() {
  93   int i;
  94   const char* formats[2] = {"%018{thread_num}", "%018n"};
  95   for (i = 0; i < sizeof(formats)/sizeof(formats[0]); ++i) {
  96     omp_set_affinity_format(formats[i]);
  97     #pragma omp parallel num_threads(8)
  98     {
  99       int j;
 100       int tid = omp_get_thread_num();
 101       char ctid = '0' + (char)tid;
 102       char* s = get_string(18);
 103       check(s[0] == ctid);
 104       for (j = 1; j < 18; ++j) {
 105         check(s[j] == ' ');
 106       }
 107     }
 108   }
 109 }
 110
 111 int main(int argc, char** argv) {
 112   check_thread_num_ljustified();
 113   check_thread_num_rjustified();
 114   check_thread_num_padded_ljustified();
 115   check_thread_num_padded_rjustified();
 116   return 0;
 117 }