123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- #include <omp.h>
- #include <stdint.h>
- #include <stdlib.h>
- const omp_alloctrait_t traits2[]
- = { { omp_atk_alignment, 16 },
- { omp_atk_sync_hint, omp_atv_default },
- { omp_atk_access, omp_atv_default },
- { omp_atk_pool_size, 1024 },
- { omp_atk_fallback, omp_atv_default_mem_fb },
- { omp_atk_partition, omp_atv_environment } };
- omp_alloctrait_t traits3[]
- = { { omp_atk_sync_hint, omp_atv_uncontended },
- { omp_atk_alignment, 32 },
- { omp_atk_access, omp_atv_all },
- { omp_atk_pool_size, 512 },
- { omp_atk_fallback, omp_atv_allocator_fb },
- { omp_atk_fb_data, 0 },
- { omp_atk_partition, omp_atv_default } };
- const omp_alloctrait_t traits4[]
- = { { omp_atk_alignment, 128 },
- { omp_atk_pool_size, 1024 },
- { omp_atk_fallback, omp_atv_null_fb } };
- int
- main ()
- {
- int *volatile p = (int *) omp_aligned_alloc (sizeof (int), 3 * sizeof (int), omp_default_mem_alloc);
- int *volatile q;
- int *volatile r;
- omp_alloctrait_t traits[3]
- = { { omp_atk_alignment, 64 },
- { omp_atk_fallback, omp_atv_null_fb },
- { omp_atk_pool_size, 4096 } };
- omp_allocator_handle_t a, a2;
- if ((((uintptr_t) p) % __alignof (int)) != 0)
- abort ();
- p[0] = 1;
- p[1] = 2;
- p[2] = 3;
- omp_free (p, omp_default_mem_alloc);
- p = (int *) omp_aligned_alloc (2 * sizeof (int), 2 * sizeof (int), omp_default_mem_alloc);
- if ((((uintptr_t) p) % (2 * sizeof (int))) != 0)
- abort ();
- p[0] = 1;
- p[1] = 2;
- omp_free (p, omp_null_allocator);
- omp_set_default_allocator (omp_default_mem_alloc);
- p = (int *) omp_aligned_alloc (1, sizeof (int), omp_null_allocator);
- if ((((uintptr_t) p) % __alignof (int)) != 0)
- abort ();
- p[0] = 3;
- omp_free (p, omp_get_default_allocator ());
- a = omp_init_allocator (omp_default_mem_space, 3, traits);
- if (a == omp_null_allocator)
- abort ();
- p = (int *) omp_aligned_alloc (32, 3072, a);
- if ((((uintptr_t) p) % 64) != 0)
- abort ();
- p[0] = 1;
- p[3071 / sizeof (int)] = 2;
- if (omp_aligned_alloc (8, 3072, a) != NULL)
- abort ();
- omp_free (p, a);
- p = (int *) omp_aligned_alloc (128, 3072, a);
- if ((((uintptr_t) p) % 128) != 0)
- abort ();
- p[0] = 3;
- p[3071 / sizeof (int)] = 4;
- omp_free (p, omp_null_allocator);
- omp_set_default_allocator (a);
- if (omp_get_default_allocator () != a)
- abort ();
- p = (int *) omp_aligned_alloc (64, 3072, omp_null_allocator);
- if (omp_aligned_alloc (8, 3072, omp_null_allocator) != NULL)
- abort ();
- omp_free (p, a);
- omp_destroy_allocator (a);
- a = omp_init_allocator (omp_default_mem_space,
- sizeof (traits2) / sizeof (traits2[0]),
- traits2);
- if (a == omp_null_allocator)
- abort ();
- if (traits3[5].key != omp_atk_fb_data)
- abort ();
- traits3[5].value = (uintptr_t) a;
- a2 = omp_init_allocator (omp_default_mem_space,
- sizeof (traits3) / sizeof (traits3[0]),
- traits3);
- if (a2 == omp_null_allocator)
- abort ();
- p = (int *) omp_aligned_alloc (4, 420, a2);
- if ((((uintptr_t) p) % 32) != 0)
- abort ();
- p[0] = 5;
- p[419 / sizeof (int)] = 6;
- q = (int *) omp_aligned_alloc (8, 768, a2);
- if ((((uintptr_t) q) % 16) != 0)
- abort ();
- q[0] = 7;
- q[767 / sizeof (int)] = 8;
- r = (int *) omp_aligned_alloc (8, 512, a2);
- if ((((uintptr_t) r) % 8) != 0)
- abort ();
- r[0] = 9;
- r[511 / sizeof (int)] = 10;
- omp_free (p, omp_null_allocator);
- omp_free (q, a2);
- omp_free (r, omp_null_allocator);
- omp_destroy_allocator (a2);
- omp_destroy_allocator (a);
- a = omp_init_allocator (omp_default_mem_space,
- sizeof (traits4) / sizeof (traits4[0]),
- traits4);
- if (a == omp_null_allocator)
- abort ();
- if (traits3[5].key != omp_atk_fb_data)
- abort ();
- traits3[5].value = (uintptr_t) a;
- a2 = omp_init_allocator (omp_default_mem_space,
- sizeof (traits3) / sizeof (traits3[0]),
- traits3);
- if (a2 == omp_null_allocator)
- abort ();
- omp_set_default_allocator (a2);
- #ifdef __cplusplus
- p = static_cast <int *> (omp_aligned_alloc (4, 420));
- #else
- p = (int *) omp_aligned_alloc (4, 420, omp_null_allocator);
- #endif
- if ((((uintptr_t) p) % 32) != 0)
- abort ();
- p[0] = 5;
- p[419 / sizeof (int)] = 6;
- q = (int *) omp_aligned_alloc (64, 768, omp_null_allocator);
- if ((((uintptr_t) q) % 128) != 0)
- abort ();
- q[0] = 7;
- q[767 / sizeof (int)] = 8;
- if (omp_aligned_alloc (8, 768, omp_null_allocator) != NULL)
- abort ();
- #ifdef __cplusplus
- omp_free (p);
- omp_free (q);
- omp_free (NULL);
- #else
- omp_free (p, omp_null_allocator);
- omp_free (q, omp_null_allocator);
- omp_free (NULL, omp_null_allocator);
- #endif
- omp_free (NULL, omp_null_allocator);
- omp_destroy_allocator (a2);
- omp_destroy_allocator (a);
- return 0;
- }
|