openmp/runtime/test/api/kmp_aligned_malloc.c

   1 // RUN: %libomp-compile-and-run
   2 #include <stdio.h>
   3 #include <stdint.h>
   4 #include <omp.h>
   5 #include "omp_testsuite.h"
   6
   7 int alignments[] = {64, 128, 256, 512, 1024, 2048, 4096};
   8
   9 unsigned aligned_by(uint64_t addr) {
  10     uint64_t alignment = 1;
  11     while((addr & (alignment-1)) == 0) {
  12         alignment <<= 1;
  13     }
  14     return (alignment >> 1);
  15 }
  16
  17 int test_kmp_aligned_malloc()
  18 {
  19   int err = 0;
  20   #pragma omp parallel shared(err)
  21   {
  22     int i;
  23     int* ptr;
  24     uint64_t addr;
  25     int tid = omp_get_thread_num();
  26
  27     for(i = 0; i < sizeof(alignments)/sizeof(int); i++) {
  28       int alignment = alignments[i];
  29       // allocate 64 bytes with 64-byte alignment
  30       // allocate 128 bytes with 128-byte alignment, etc.
  31       ptr = (int*)kmp_aligned_malloc(alignment, alignment);
  32       addr = (uint64_t)ptr;
  33       if(addr & (alignment-1)) {
  34         printf("thread %d: addr = %p (aligned to %u bytes) but expected "
  35                " alignment = %d\n", tid, ptr, aligned_by(addr), alignment);
  36         err = 1;
  37       }
  38       kmp_free(ptr);
  39     }
  40
  41     ptr = kmp_aligned_malloc(128, 127);
  42     if (ptr != NULL) {
  43       printf("thread %d: kmp_aligned_malloc() didn't return NULL when "
  44              "alignment was not power of 2\n", tid);
  45       err = 1;
  46     }
  47   } /* end of parallel */
  48   return !err;
  49 }
  50
  51 int main()
  52 {
  53   int i;
  54   int num_failed=0;
  55
  56   for(i = 0; i < REPETITIONS; i++) {
  57     if(!test_kmp_aligned_malloc()) {
  58       num_failed++;
  59     }
  60   }
  61   return num_failed;
  62 }