mirror of
https://gcc.gnu.org/git/gcc.git
synced 2024-11-27 05:44:15 +08:00
openmp: Add alloc_align attribute to omp_aligned_*alloc and testcase for omp_realloc
This patch adds alloc_align attribute to omp_aligned_{,c}alloc so that if
the first argument is constant, GCC can assume requested alignment.
Additionally, it adds testsuite coverage for omp_realloc which I haven't
managed to write in the patch from yesterday.
2021-10-01 Jakub Jelinek <jakub@redhat.com>
* omp.h.in (omp_aligned_alloc, omp_aligned_calloc): Add
__alloc_align__ (1) attribute.
* testsuite/libgomp.c-c++-common/alloc-9.c: New test.
This commit is contained in:
Jakub Jelinek
parent
701075864a
commit
998e434f8f
@ -306,7 +306,7 @@ extern void *omp_aligned_alloc (__SIZE_TYPE__, __SIZE_TYPE__,
|
||||
omp_allocator_handle_t
|
||||
__GOMP_DEFAULT_NULL_ALLOCATOR)
|
||||
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
||||
__alloc_size__ (2)));
|
||||
__alloc_size__ (2), __alloc_align__ (1)));
|
||||
extern void *omp_calloc (__SIZE_TYPE__, __SIZE_TYPE__,
|
||||
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
||||
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
||||
@ -315,7 +315,7 @@ extern void *omp_aligned_calloc (__SIZE_TYPE__, __SIZE_TYPE__, __SIZE_TYPE__,
|
||||
omp_allocator_handle_t
|
||||
__GOMP_DEFAULT_NULL_ALLOCATOR)
|
||||
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
||||
__alloc_size__ (2, 3)));
|
||||
__alloc_size__ (2, 3), __alloc_align__ (1)));
|
||||
extern void *omp_realloc (void *, __SIZE_TYPE__,
|
||||
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR,
|
||||
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
||||
|
||||
271
libgomp/testsuite/libgomp.c-c++-common/alloc-9.c
Normal file
271
libgomp/testsuite/libgomp.c-c++-common/alloc-9.c
Normal file
@ -0,0 +1,271 @@
|
||||
#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_alloc (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_alloctrait_t traits5[2]
|
||||
= { { 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;
|
||||
p = (int *) omp_realloc (p, 4 * sizeof (int), omp_default_mem_alloc, omp_default_mem_alloc);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0 || p[0] != 1 || p[1] != 2 || p[2] != 3)
|
||||
abort ();
|
||||
p[0] = 4;
|
||||
p[1] = 5;
|
||||
p[2] = 6;
|
||||
p[3] = 7;
|
||||
p = (int *) omp_realloc (p, 2 * sizeof (int), omp_default_mem_alloc, omp_default_mem_alloc);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0 || p[0] != 4 || p[1] != 5)
|
||||
abort ();
|
||||
p[0] = 8;
|
||||
p[1] = 9;
|
||||
if (omp_realloc (p, 0, omp_null_allocator, omp_default_mem_alloc) != NULL)
|
||||
abort ();
|
||||
p = (int *) omp_realloc (NULL, 2 * sizeof (int), omp_default_mem_alloc, omp_null_allocator);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0)
|
||||
abort ();
|
||||
p[0] = 1;
|
||||
p[1] = 2;
|
||||
p = (int *) omp_realloc (p, 5 * sizeof (int), omp_default_mem_alloc, omp_default_mem_alloc);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0 || p[0] != 1 || p[1] != 2)
|
||||
abort ();
|
||||
p[0] = 3;
|
||||
p[1] = 4;
|
||||
p[2] = 5;
|
||||
p[3] = 6;
|
||||
p[4] = 7;
|
||||
omp_free (p, omp_null_allocator);
|
||||
omp_set_default_allocator (omp_default_mem_alloc);
|
||||
if (omp_realloc (NULL, 0, omp_null_allocator, omp_null_allocator) != NULL)
|
||||
abort ();
|
||||
p = (int *) omp_alloc (sizeof (int), omp_null_allocator);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0)
|
||||
abort ();
|
||||
p[0] = 3;
|
||||
p = (int *) omp_realloc (p, 3 * sizeof (int), omp_null_allocator, omp_null_allocator);
|
||||
if ((((uintptr_t) p) % __alignof (int)) != 0 || p[0] != 3)
|
||||
abort ();
|
||||
p[0] = 4;
|
||||
p[1] = 5;
|
||||
p[2] = 6;
|
||||
if (omp_realloc (p, 0, omp_null_allocator, omp_get_default_allocator ()) != NULL)
|
||||
abort ();
|
||||
a = omp_init_allocator (omp_default_mem_space, 3, traits);
|
||||
if (a == omp_null_allocator)
|
||||
abort ();
|
||||
p = (int *) omp_alloc (sizeof (int), a);
|
||||
if ((((uintptr_t) p) % 64) != 0)
|
||||
abort ();
|
||||
p[0] = 7;
|
||||
p = (int *) omp_realloc (p, 3072, a, a);
|
||||
if ((((uintptr_t) p) % 64) != 0 || p[0] != 7)
|
||||
abort ();
|
||||
p[0] = 1;
|
||||
p[3071 / sizeof (int)] = 2;
|
||||
q = (int *) omp_alloc (sizeof (int), a);
|
||||
if ((((uintptr_t) q) % 64) != 0)
|
||||
abort ();
|
||||
q[0] = 8;
|
||||
if (omp_realloc (q, 3072, a, a) != NULL)
|
||||
abort ();
|
||||
omp_free (p, a);
|
||||
omp_free (q, a);
|
||||
p = (int *) omp_alloc (sizeof (int), a);
|
||||
p[0] = 42;
|
||||
p = (int *) omp_realloc (p, 3072, a, a);
|
||||
if (p[0] != 42)
|
||||
abort ();
|
||||
p[0] = 3;
|
||||
p[3071 / sizeof (int)] = 4;
|
||||
omp_realloc (p, 0, omp_null_allocator, omp_null_allocator);
|
||||
omp_set_default_allocator (a);
|
||||
if (omp_get_default_allocator () != a)
|
||||
abort ();
|
||||
p = (int *) omp_alloc (31, omp_null_allocator);
|
||||
if (p == NULL)
|
||||
abort ();
|
||||
p = (int *) omp_realloc (p, 3072, omp_null_allocator, omp_null_allocator);
|
||||
if (p == NULL)
|
||||
abort ();
|
||||
q = (int *) omp_alloc (sizeof (int), omp_null_allocator);
|
||||
if (q == NULL)
|
||||
abort ();
|
||||
if (omp_realloc (q, 3072, omp_null_allocator, omp_null_allocator) != NULL)
|
||||
abort ();
|
||||
omp_free (p, a);
|
||||
omp_free (q, a);
|
||||
omp_destroy_allocator (a);
|
||||
|
||||
a = omp_init_allocator (omp_default_mem_space, 2, traits5);
|
||||
if (a == omp_null_allocator)
|
||||
abort ();
|
||||
omp_set_default_allocator (a);
|
||||
if (omp_get_default_allocator () != a)
|
||||
abort ();
|
||||
p = (int *) omp_alloc (3071, omp_null_allocator);
|
||||
if (p == NULL)
|
||||
abort ();
|
||||
p = (int *) omp_realloc (p, 3072, omp_null_allocator, omp_null_allocator);
|
||||
if (p == NULL)
|
||||
abort ();
|
||||
q = (int *) omp_alloc (sizeof (int), omp_null_allocator);
|
||||
if (q == NULL)
|
||||
abort ();
|
||||
if (omp_realloc (q, 3072, omp_null_allocator, omp_null_allocator) != NULL)
|
||||
abort ();
|
||||
omp_free (p, a);
|
||||
omp_free (q, 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_alloc (sizeof (int), a2);
|
||||
if ((((uintptr_t) p) % 32) != 0)
|
||||
abort ();
|
||||
p[0] = 84;
|
||||
p = (int *) omp_realloc (p, 380, a2, a2);
|
||||
if ((((uintptr_t) p) % 32) != 0 || p[0] != 84)
|
||||
abort ();
|
||||
p[0] = 5;
|
||||
p[379 / sizeof (int)] = 6;
|
||||
q = (int *) omp_alloc (sizeof (int), a2);
|
||||
if ((((uintptr_t) q) % 32) != 0)
|
||||
abort ();
|
||||
q[0] = 42;
|
||||
q = (int *) omp_realloc (q, 768, a2, a2);
|
||||
if ((((uintptr_t) q) % 16) != 0 || q[0] != 42)
|
||||
abort ();
|
||||
q[0] = 7;
|
||||
q[767 / sizeof (int)] = 8;
|
||||
r = (int *) omp_realloc (NULL, 512, a2, omp_null_allocator);
|
||||
if ((((uintptr_t) r) % __alignof (int)) != 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);
|
||||
p = (int *) omp_alloc (sizeof (int), a2);
|
||||
if ((((uintptr_t) p) % 32) != 0)
|
||||
abort ();
|
||||
p[0] = 85;
|
||||
p = (int *) omp_realloc (p, 420, a, a2);
|
||||
if ((((uintptr_t) p) % 16) != 0 || p[0] != 85)
|
||||
abort ();
|
||||
p[0] = 5;
|
||||
p[419 / sizeof (int)] = 6;
|
||||
q = (int *) omp_alloc (sizeof (int), a);
|
||||
if ((((uintptr_t) q) % 16) != 0)
|
||||
abort ();
|
||||
q[0] = 43;
|
||||
q = (int *) omp_realloc (q, 420, a2, a);
|
||||
if ((((uintptr_t) q) % 32) != 0 || q[0] != 43)
|
||||
abort ();
|
||||
q[0] = 44;
|
||||
q[419 / sizeof (int)] = 8;
|
||||
q = (int *) omp_realloc (q, 768, a2, a2);
|
||||
if ((((uintptr_t) q) % 16) != 0 || q[0] != 44)
|
||||
abort ();
|
||||
q[0] = 7;
|
||||
q[767 / sizeof (int)] = 8;
|
||||
omp_free (p, omp_null_allocator);
|
||||
omp_free (q, a2);
|
||||
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_realloc (NULL, 420));
|
||||
#else
|
||||
p = (int *) omp_realloc (NULL, 420, omp_null_allocator, omp_null_allocator);
|
||||
#endif
|
||||
if ((((uintptr_t) p) % 32) != 0)
|
||||
abort ();
|
||||
p[0] = 5;
|
||||
p[419 / sizeof (int)] = 6;
|
||||
q = (int *) omp_realloc (NULL, sizeof (int), omp_null_allocator, omp_null_allocator);
|
||||
if ((((uintptr_t) q) % 32) != 0)
|
||||
abort ();
|
||||
q[0] = 99;
|
||||
q = (int *) omp_realloc (q, 700, omp_null_allocator, omp_null_allocator);
|
||||
if ((((uintptr_t) q) % 128) != 0 || q[0] != 99)
|
||||
abort ();
|
||||
q[0] = 7;
|
||||
q[699 / sizeof (int)] = 8;
|
||||
if (omp_realloc (NULL, 768, omp_null_allocator, omp_null_allocator) != NULL)
|
||||
abort ();
|
||||
#ifdef __cplusplus
|
||||
omp_free (p);
|
||||
if (omp_realloc (q, 0) != NULL)
|
||||
abort ();
|
||||
omp_free (NULL);
|
||||
#else
|
||||
omp_free (p, omp_null_allocator);
|
||||
if (omp_realloc (q, 0, omp_null_allocator, omp_null_allocator) != NULL)
|
||||
abort ();
|
||||
omp_free (NULL, omp_null_allocator);
|
||||
#endif
|
||||
omp_free (NULL, omp_null_allocator);
|
||||
omp_destroy_allocator (a2);
|
||||
omp_destroy_allocator (a);
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user