reference, declaration → definition
definition → references, declarations, derived classes, virtual overrides
reference to multiple definitions → definitions
unreferenced

    1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62

// RUN: %libomp-compile-and-run
#include <stdio.h>
#include <stdint.h>
#include <omp.h>
#include "omp_testsuite.h"

int alignments[] = {64, 128, 256, 512, 1024, 2048, 4096};

unsigned aligned_by(uint64_t addr) {
    uint64_t alignment = 1;
    while((addr & (alignment-1)) == 0) {
        alignment <<= 1;
    }
    return (alignment >> 1);
}

int test_kmp_aligned_malloc()
{
  int err = 0;
  #pragma omp parallel shared(err)
  {
    int i;
    int* ptr;
    uint64_t addr;
    int tid = omp_get_thread_num();

    for(i = 0; i < sizeof(alignments)/sizeof(int); i++) {
      int alignment = alignments[i];
      // allocate 64 bytes with 64-byte alignment
      // allocate 128 bytes with 128-byte alignment, etc.
      ptr = (int*)kmp_aligned_malloc(alignment, alignment);
      addr = (uint64_t)ptr;
      if(addr & (alignment-1)) {
        printf("thread %d: addr = %p (aligned to %u bytes) but expected "
               " alignment = %d\n", tid, ptr, aligned_by(addr), alignment);
        err = 1;
      }
      kmp_free(ptr);
    }

    ptr = kmp_aligned_malloc(128, 127);
    if (ptr != NULL) {
      printf("thread %d: kmp_aligned_malloc() didn't return NULL when "
             "alignment was not power of 2\n", tid);
      err = 1;
    }
  } /* end of parallel */
  return !err;
}

int main()
{
  int i;
  int num_failed=0;

  for(i = 0; i < REPETITIONS; i++) {
    if(!test_kmp_aligned_malloc()) {
      num_failed++;
    }
  }
  return num_failed;
}