diff options
author | Kwok Cheung Yeung <kcy@codesourcery.com> | 2021-03-01 14:15:30 -0800 |
---|---|---|
committer | Kwok Cheung Yeung <kcy@codesourcery.com> | 2021-03-05 04:18:50 -0800 |
commit | 58def69b6d68276745e5cc14d6ca24ad47be5d27 (patch) | |
tree | 0fd240121f42e1d2a6353a3c33f47c549681c3fb | |
parent | 1c3f38b30c1db0aef5ccbf6d20fb5fd13785d482 (diff) | |
download | gcc-58def69b6d68276745e5cc14d6ca24ad47be5d27.zip gcc-58def69b6d68276745e5cc14d6ca24ad47be5d27.tar.gz gcc-58def69b6d68276745e5cc14d6ca24ad47be5d27.tar.bz2 |
openmp: Scale type precision of collapsed iterator variable
This sets the type precision of the collapsed iterator variable to the
sum of the precision of the collapsed loop variables, up to a maximum of
sizeof(long long) (i.e. 64-bits).
2021-03-01 Kwok Cheung Yeung <kcy@codesourcery.com>
gcc/
* omp-expand.c (expand_oacc_for): Convert .tile variable to
diff_type before multiplying.
* omp-general.c (omp_extract_for_data): Use accumulated precision
of all collapsed for-loops as precision of iteration variable, up
to the precision of a long long.
libgomp/
* testsuite/libgomp.c-c++-common/collapse-4.c: New.
* testsuite/libgomp.fortran/collapse5.f90: New.
-rw-r--r-- | gcc/ChangeLog.omp | 8 | ||||
-rw-r--r-- | gcc/omp-expand.c | 5 | ||||
-rw-r--r-- | gcc/omp-general.c | 29 | ||||
-rw-r--r-- | libgomp/ChangeLog.omp | 5 | ||||
-rw-r--r-- | libgomp/testsuite/libgomp.c-c++-common/collapse-4.c | 23 | ||||
-rw-r--r-- | libgomp/testsuite/libgomp.fortran/collapse5.f90 | 23 |
6 files changed, 86 insertions, 7 deletions
diff --git a/gcc/ChangeLog.omp b/gcc/ChangeLog.omp index a59c25b..374665d 100644 --- a/gcc/ChangeLog.omp +++ b/gcc/ChangeLog.omp @@ -1,3 +1,11 @@ +2021-03-01 Kwok Cheung Yeung <kcy@codesourcery.com> + + * omp-expand.c (expand_oacc_for): Convert .tile variable to + diff_type before multiplying. + * omp-general.c (omp_extract_for_data): Use accumulated precision + of all collapsed for-loops as precision of iteration variable, up + to the precision of a long long. + 2021-02-24 Julian Brown <julian@codesourcery.com> Backport from mainline diff --git a/gcc/omp-expand.c b/gcc/omp-expand.c index e4a2f3a..f8347c0 100644 --- a/gcc/omp-expand.c +++ b/gcc/omp-expand.c @@ -7328,7 +7328,10 @@ expand_oacc_for (struct omp_region *region, struct omp_for_data *fd) tile_size = create_tmp_var (diff_type, ".tile_size"); expr = build_int_cst (diff_type, 1); for (int ix = 0; ix < fd->collapse; ix++) - expr = fold_build2 (MULT_EXPR, diff_type, counts[ix].tile, expr); + { + tree tile = fold_convert (diff_type, counts[ix].tile); + expr = fold_build2 (MULT_EXPR, diff_type, tile, expr); + } expr = force_gimple_operand_gsi (&gsi, expr, true, NULL_TREE, true, GSI_SAME_STMT); ass = gimple_build_assign (tile_size, expr); diff --git a/gcc/omp-general.c b/gcc/omp-general.c index 8e5b961..97f94e1 100644 --- a/gcc/omp-general.c +++ b/gcc/omp-general.c @@ -356,6 +356,7 @@ omp_extract_for_data (gomp_for *for_stmt, struct omp_for_data *fd, fd->non_rect = true; } } + int accum_iter_precision = 0; for (i = 0; i < cnt; i++) { if (i == 0 @@ -438,12 +439,28 @@ omp_extract_for_data (gomp_for *for_stmt, struct omp_for_data *fd, { if (fd->collapse == 1 && !fd->tiling) iter_type = TREE_TYPE (loop->v); - else if (i == 0 - || TYPE_PRECISION (iter_type) - < TYPE_PRECISION (TREE_TYPE (loop->v))) - iter_type - = build_nonstandard_integer_type - (TYPE_PRECISION (TREE_TYPE (loop->v)), 1); + else + { + int loop_precision = TYPE_PRECISION (TREE_TYPE (loop->v)); + int iter_type_precision = 0; + const int max_accum_precision + = TYPE_PRECISION (long_long_unsigned_type_node); + + accum_iter_precision += loop_precision; + + if (i == 0 + || (loop_precision >= max_accum_precision + && loop_precision >= TYPE_PRECISION (iter_type))) + iter_type_precision = loop_precision; + else if (TYPE_PRECISION (iter_type) < max_accum_precision) + iter_type_precision + = MIN (1 << ceil_log2 (accum_iter_precision), + max_accum_precision); + + if (iter_type_precision) + iter_type = build_nonstandard_integer_type + (iter_type_precision, 1); + } } else if (iter_type != long_long_unsigned_type_node) { diff --git a/libgomp/ChangeLog.omp b/libgomp/ChangeLog.omp index d1dcf20..0e3fd12 100644 --- a/libgomp/ChangeLog.omp +++ b/libgomp/ChangeLog.omp @@ -1,3 +1,8 @@ +2021-03-01 Kwok Cheung Yeung <kcy@codesourcery.com> + + * testsuite/libgomp.c-c++-common/collapse-4.c: New. + * testsuite/libgomp.fortran/collapse5.f90: New. + 2021-02-24 Julian Brown <julian@codesourcery.com> Backport from mainline diff --git a/libgomp/testsuite/libgomp.c-c++-common/collapse-4.c b/libgomp/testsuite/libgomp.c-c++-common/collapse-4.c new file mode 100644 index 0000000..c0af29f --- /dev/null +++ b/libgomp/testsuite/libgomp.c-c++-common/collapse-4.c @@ -0,0 +1,23 @@ +/* { dg-do run } */ + +#include <stdlib.h> + +int +main (void) +{ + int i, j; + int count = 0; + + #pragma omp parallel for collapse(2) + for (i = 0; i < 80000; i++) + for (j = 0; j < 80000; j++) + if (i == 66666 && j == 77777) + /* In the collapsed loop space, this is iteration + 66666*80000+77777==5,333,357,777. If the type of the iterator + for the collapsed loop is only a 32-bit unsigned int, then this + iteration will exceed its maximum range and be skipped. */ + count++; + + if (count != 1) + abort (); +} diff --git a/libgomp/testsuite/libgomp.fortran/collapse5.f90 b/libgomp/testsuite/libgomp.fortran/collapse5.f90 new file mode 100644 index 0000000..5632d9b --- /dev/null +++ b/libgomp/testsuite/libgomp.fortran/collapse5.f90 @@ -0,0 +1,23 @@ +! { dg-do run } + +program collapse5 + implicit none + + integer :: i, j + integer :: count = 0 + + !$omp parallel do collapse (2) + do i = 1, 80000 + do j = 1, 80000 + if (i .eq. 66666 .and. j .eq. 77777) then + ! In the collapsed loop space, this is iteration + ! 66666*80000+77777==5,333,357,777. If the type of the iterator + ! for the collapsed loop is only a 32-bit unsigned int, then this + ! iteration will exceed its maximum range and be skipped. + count = count + 1 + end if + end do + end do + + if (count .ne. 1) stop 1 +end |