openmp/runtime/test/affinity/kmp-affinity-reset.c

   1 // RUN: %libomp-compile -D_GNU_SOURCE
   2 // RUN: env OMP_NUM_THREADS=2,2 KMP_AFFINITY=reset,granularity=thread,compact %libomp-run
   3 // REQUIRES: linux
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8 #include <omp.h>
   9 #include "libomp_test_affinity.h"
  10
  11 #define CHECK_EQUAL 0
  12 #define CHECK_NOT_EQUAL 1
  13
  14 void check_primary_thread_affinity(int line, affinity_mask_t *other_aff,
  15                                    int type) {
  16   #pragma omp master
  17   {
  18     affinity_mask_t *primary_aff = affinity_mask_alloc();
  19     get_thread_affinity(primary_aff);
  20     if (type == CHECK_EQUAL && !affinity_mask_equal(primary_aff, other_aff)) {
  21       fprintf(stderr, "error: line %d: primary affinity was not equal\n", line);
  22       exit(EXIT_FAILURE);
  23     } else if (type == CHECK_NOT_EQUAL &&
  24                affinity_mask_equal(primary_aff, other_aff)) {
  25       fprintf(stderr, "error: line %d: primary affinity was equal\n", line);
  26       exit(EXIT_FAILURE);
  27     }
  28     affinity_mask_free(primary_aff);
  29   }
  30 }
  31
  32 #define CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(other_aff)                         \
  33   check_primary_thread_affinity(__LINE__, other_aff, CHECK_EQUAL)
  34 #define CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(other_aff)                     \
  35   check_primary_thread_affinity(__LINE__, other_aff, CHECK_NOT_EQUAL)
  36
  37 int main() {
  38   int i;
  39   affinity_mask_t *initial_mask = affinity_mask_alloc();
  40   get_thread_affinity(initial_mask);
  41
  42   for (i = 0; i < 10; ++i) {
  43     #pragma omp parallel
  44     {
  45       CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
  46     }
  47     CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
  48   }
  49
  50   omp_set_max_active_levels(2);
  51   for (i = 0; i < 10; ++i) {
  52     #pragma omp parallel
  53     {
  54       CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
  55
  56       #pragma omp parallel
  57       CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
  58
  59       CHECK_PRIMARY_THREAD_AFFINITY_NOT_EQUAL(initial_mask);
  60     }
  61     CHECK_PRIMARY_THREAD_AFFINITY_EQUAL(initial_mask);
  62   }
  63
  64   affinity_mask_free(initial_mask);
  65   return EXIT_SUCCESS;
  66 }