1  #include <omp.h>
       2  #include <stdint.h>
       3  #include <stdlib.h>
       4  
       5  const omp_alloctrait_t traits2[]
       6  = { { omp_atk_alignment, 16 },
       7      { omp_atk_sync_hint, omp_atv_default },
       8      { omp_atk_access, omp_atv_default },
       9      { omp_atk_pool_size, 1024 },
      10      { omp_atk_fallback, omp_atv_default_mem_fb },
      11      { omp_atk_partition, omp_atv_environment } };
      12  omp_alloctrait_t traits3[]
      13  = { { omp_atk_sync_hint, omp_atv_uncontended },
      14      { omp_atk_alignment, 32 },
      15      { omp_atk_access, omp_atv_all },
      16      { omp_atk_pool_size, 512 },
      17      { omp_atk_fallback, omp_atv_allocator_fb },
      18      { omp_atk_fb_data, 0 },
      19      { omp_atk_partition, omp_atv_default } };
      20  const omp_alloctrait_t traits4[]
      21  = { { omp_atk_alignment, 128 },
      22      { omp_atk_pool_size, 1024 },
      23      { omp_atk_fallback, omp_atv_null_fb } };
      24  
      25  int
      26  main ()
      27  {
      28    int *volatile p = (int *) omp_alloc (3 * sizeof (int), omp_default_mem_alloc);
      29    int *volatile q;
      30    int *volatile r;
      31    omp_alloctrait_t traits[3]
      32      = { { omp_atk_alignment, 64 },
      33  	{ omp_atk_fallback, omp_atv_null_fb },
      34  	{ omp_atk_pool_size, 4096 } };
      35    omp_allocator_handle_t a, a2;
      36  
      37    if ((((uintptr_t) p) % __alignof (int)) != 0)
      38      abort ();
      39    p[0] = 1;
      40    p[1] = 2;
      41    p[2] = 3;
      42    omp_free (p, omp_default_mem_alloc);
      43    p = (int *) omp_alloc (2 * sizeof (int), omp_default_mem_alloc);
      44    if ((((uintptr_t) p) % __alignof (int)) != 0)
      45      abort ();
      46    p[0] = 1;
      47    p[1] = 2;
      48    omp_free (p, omp_null_allocator);
      49    omp_set_default_allocator (omp_default_mem_alloc);
      50    p = (int *) omp_alloc (sizeof (int), omp_null_allocator);
      51    if ((((uintptr_t) p) % __alignof (int)) != 0)
      52      abort ();
      53    p[0] = 3;
      54    omp_free (p, omp_get_default_allocator ());
      55  
      56    a = omp_init_allocator (omp_default_mem_space, 3, traits);
      57    if (a == omp_null_allocator)
      58      abort ();
      59    p = (int *) omp_alloc (3072, a);
      60    if ((((uintptr_t) p) % 64) != 0)
      61      abort ();
      62    p[0] = 1;
      63    p[3071 / sizeof (int)] = 2;
      64    if (omp_alloc (3072, a) != NULL)
      65      abort ();
      66    omp_free (p, a);
      67    p = (int *) omp_alloc (3072, a);
      68    p[0] = 3;
      69    p[3071 / sizeof (int)] = 4;
      70    omp_free (p, omp_null_allocator);
      71    omp_set_default_allocator (a);
      72    if (omp_get_default_allocator () != a)
      73      abort ();
      74    p = (int *) omp_alloc (3072, omp_null_allocator);
      75    if (omp_alloc (3072, omp_null_allocator) != NULL)
      76      abort ();
      77    omp_free (p, a);
      78    omp_destroy_allocator (a);
      79  
      80    a = omp_init_allocator (omp_default_mem_space,
      81  			  sizeof (traits2) / sizeof (traits2[0]),
      82  			  traits2);
      83    if (a == omp_null_allocator)
      84      abort ();
      85    if (traits3[5].key != omp_atk_fb_data)
      86      abort ();
      87    traits3[5].value = (uintptr_t) a;
      88    a2 = omp_init_allocator (omp_default_mem_space,
      89  			   sizeof (traits3) / sizeof (traits3[0]),
      90  			   traits3);
      91    if (a2 == omp_null_allocator)
      92      abort ();
      93    p = (int *) omp_alloc (420, a2);
      94    if ((((uintptr_t) p) % 32) != 0)
      95      abort ();
      96    p[0] = 5;
      97    p[419 / sizeof (int)] = 6;
      98    q = (int *) omp_alloc (768, a2);
      99    if ((((uintptr_t) q) % 16) != 0)
     100      abort ();
     101    q[0] = 7;
     102    q[767 / sizeof (int)] = 8;
     103    r = (int *) omp_alloc (512, a2);
     104    if ((((uintptr_t) r) % __alignof (int)) != 0)
     105      abort ();
     106    r[0] = 9;
     107    r[511 / sizeof (int)] = 10;
     108    omp_free (p, omp_null_allocator);
     109    omp_free (q, a2);
     110    omp_free (r, omp_null_allocator);
     111    omp_destroy_allocator (a2);
     112    omp_destroy_allocator (a);
     113  
     114    a = omp_init_allocator (omp_default_mem_space,
     115  			  sizeof (traits4) / sizeof (traits4[0]),
     116  			  traits4);
     117    if (a == omp_null_allocator)
     118      abort ();
     119    if (traits3[5].key != omp_atk_fb_data)
     120      abort ();
     121    traits3[5].value = (uintptr_t) a;
     122    a2 = omp_init_allocator (omp_default_mem_space,
     123  			   sizeof (traits3) / sizeof (traits3[0]),
     124  			   traits3);
     125    if (a2 == omp_null_allocator)
     126      abort ();
     127    omp_set_default_allocator (a2);
     128  #ifdef __cplusplus
     129    p = static_cast <int *> (omp_alloc (420));
     130  #else
     131    p = (int *) omp_alloc (420, omp_null_allocator);
     132  #endif
     133    if ((((uintptr_t) p) % 32) != 0)
     134      abort ();
     135    p[0] = 5;
     136    p[419 / sizeof (int)] = 6;
     137    q = (int *) omp_alloc (768, omp_null_allocator);
     138    if ((((uintptr_t) q) % 128) != 0)
     139      abort ();
     140    q[0] = 7;
     141    q[767 / sizeof (int)] = 8;
     142    if (omp_alloc (768, omp_null_allocator) != NULL)
     143      abort ();
     144  #ifdef __cplusplus
     145    omp_free (p);
     146    omp_free (q);
     147    omp_free (NULL);
     148  #else
     149    omp_free (p, omp_null_allocator);
     150    omp_free (q, omp_null_allocator);
     151    omp_free (NULL, omp_null_allocator);
     152  #endif
     153    omp_free (NULL, omp_null_allocator);
     154    omp_destroy_allocator (a2);
     155    omp_destroy_allocator (a);
     156    return 0;
     157  }