diff mbox series

nvptx: Add support for subword compare-and-swap

Message ID 3cbe58e6-b427-ff0d-b7d1-8723aac82b18@codesourcery.com
State New
Headers show
Series nvptx: Add support for subword compare-and-swap | expand

Commit Message

Kwok Cheung Yeung June 15, 2020, 8:28 p.m. UTC
Hello

This patch adds support on nvptx for __sync_val_compare_and_swap operations on 
1- and 2-byte values. The implementation is a straight copy of the version for 
AMD GCN.

I have added a new libgomp test that exercises the new operation. I have also 
verified that the new code does not cause any regressions on the nvptx 
offloading tests, and that the new test passes with both nvptx and amdgcn as 
offload targets.

Okay for master and OG10?

Kwok
commit 7c3a9c23ba9f5b8fe953aa5492ae75617f2444a3
Author: Kwok Cheung Yeung <kcy@codesourcery.com>
Date:   Mon Jun 15 12:34:55 2020 -0700

    nvptx: Add support for subword compare-and-swap
    
    2020-06-15  Kwok Cheung Yeung  <kcy@codesourcery.com>
    
    	libgcc/
    	* config/nvptx/atomic.c: New.
    	* config/nvptx/t-nvptx (LIB2ADD): Add atomic.c.
    
    	libgomp/
    	* testsuite/libgomp.c-c++-common/reduction-16.c: New.

Comments

Thomas Schwinge June 23, 2020, 4:44 p.m. UTC | #1
Hi!

On 2020-06-15T21:28:12+0100, Kwok Cheung Yeung <kcy@codesourcery.com> wrote:
> This patch adds support on nvptx for __sync_val_compare_and_swap operations on
> 1- and 2-byte values.

Is this a thorough review that these are the only functions missing, or
did you just implement what you found missing for some test case you've
been looking into?  Other architectures' similar libgcc files seem to be
defining more of such related functions.

Per the PTX 3.1 manual that I looked into, I see for CAS it supports:
'.b32', '.b64'.  We've got:

    $ build-gcc-offload-nvptx-none/gcc/xgcc -Bbuild-gcc-offload-nvptx-none/gcc -dM -E -x c /dev/null | grep -i compare.and.swap
    #define __GCC_HAVE_SYNC_COMPARE_AND_SWAP_4 1
    #define __GCC_HAVE_SYNC_COMPARE_AND_SWAP_8 1

..., so that would match the PTX 3.1 manual.  GCC also seems to know
about 16-byte ("'.b128'", which doesn't exist in PTX).  A quick run of
your testcase with 'GENERATE_TEST(__int128)' seems to just work, but I
haven't verified why/how.

> The implementation is a straight copy of the version for
> AMD GCN.

(Should thus be generalized?  That can be done later, as far as I'm
concerned -- there already seems to be quite some code duplication in
libgcc.)

I have not verified the algorithm.

It seems a bit unfortunate to have such a thing outlined in a separate
function, given we're talking about performance-critical code here?  Even
more so for GCN, where there's no JIT compiler that can inline it later,
as it's the case for nvptx?

You have verified that GCC itself shouldn't/can't synthesize such
replacement code, inline?

The GCN/nvptx libgcc code actually seems simple enough so that GCC could
synthesize that internally -- or is there a reason not to?  (Just
curious.)

I see 'gcc/doc/generic.texi', "OMP_ATOMIC" state:

    The gimplifier tries
    three alternative code generation strategies.  Whenever possible,
    an atomic update built-in is used.  If that fails, a
    compare-and-swap loop is attempted.  If that also fails, a
    regular critical section around the expression is used.

..., and I see 'gcc/omp-expand.c:expand_omp_atomic_pipeline' synthesize
that "compare-and-swap loop" code, which looks vaguely similar to your
libgcc implementation.

..., or maybe even 'gcc/builtins.c:expand_builtin_compare_and_swap'
etc. could be doing such things?

..., and 'gcc/optabs.c:expand_compare_and_swap_loop' etc. also look
similar/related?

I clearly don't know the history behind all of that.  :-|

> I have added a new libgomp test that exercises the new operation.

Ideally we should also have nvptx target compile-time test, and scan the
PTX assembler code generated -- but we don't have that for a lot of
things, so this is probably OK without, too?

> I have also
> verified that the new code does not cause any regressions on the nvptx
> offloading tests, and that the new test passes with both nvptx and amdgcn as
> offload targets.
>
> Okay for master and OG10?

Given that this solves an actual problem, and we have precedence with the
GCN target doing the same thing, this seems OK (but I can't formally
approve).


Grüße
 Thomas


> commit 7c3a9c23ba9f5b8fe953aa5492ae75617f2444a3
> Author: Kwok Cheung Yeung <kcy@codesourcery.com>
> Date:   Mon Jun 15 12:34:55 2020 -0700
>
>     nvptx: Add support for subword compare-and-swap
>
>     2020-06-15  Kwok Cheung Yeung  <kcy@codesourcery.com>
>
>       libgcc/
>       * config/nvptx/atomic.c: New.
>       * config/nvptx/t-nvptx (LIB2ADD): Add atomic.c.
>
>       libgomp/
>       * testsuite/libgomp.c-c++-common/reduction-16.c: New.
>
> diff --git a/libgcc/config/nvptx/atomic.c b/libgcc/config/nvptx/atomic.c
> new file mode 100644
> index 0000000..4becbd2
> --- /dev/null
> +++ b/libgcc/config/nvptx/atomic.c
> @@ -0,0 +1,59 @@
> +/* NVPTX atomic operations
> +   Copyright (C) 2020 Free Software Foundation, Inc.
> +   Contributed by Mentor Graphics.
> +
> +   This file is free software; you can redistribute it and/or modify it
> +   under the terms of the GNU General Public License as published by the
> +   Free Software Foundation; either version 3, or (at your option) any
> +   later version.
> +
> +   This file is distributed in the hope that it will be useful, but
> +   WITHOUT ANY WARRANTY; without even the implied warranty of
> +   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
> +   General Public License for more details.
> +
> +   Under Section 7 of GPL version 3, you are granted additional
> +   permissions described in the GCC Runtime Library Exception, version
> +   3.1, as published by the Free Software Foundation.
> +
> +   You should have received a copy of the GNU General Public License and
> +   a copy of the GCC Runtime Library Exception along with this program;
> +   see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
> +   <http://www.gnu.org/licenses/>.  */
> +
> +#include <stdbool.h>
> +
> +#define __SYNC_SUBWORD_COMPARE_AND_SWAP(TYPE, SIZE)                       \
> +                                                                          \
> +TYPE                                                                      \
> +__sync_val_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)     \
> +{                                                                         \
> +  unsigned int *wordptr = (unsigned int *)((__UINTPTR_TYPE__ ) ptr & ~3UL);  \
> +  int shift = ((__UINTPTR_TYPE__ ) ptr & 3UL) * 8;                        \
> +  unsigned int valmask = (1 << (SIZE * 8)) - 1;                                   \
> +  unsigned int wordmask = ~(valmask << shift);                                    \
> +  unsigned int oldword = *wordptr;                                        \
> +  for (;;)                                                                \
> +    {                                                                             \
> +      TYPE prevval = (oldword >> shift) & valmask;                        \
> +      if (__builtin_expect (prevval != oldval, 0))                        \
> +     return prevval;                                                      \
> +      unsigned int newword = oldword & wordmask;                          \
> +      newword |= ((unsigned int) newval) << shift;                        \
> +      unsigned int prevword                                               \
> +       = __sync_val_compare_and_swap_4 (wordptr, oldword, newword);       \
> +      if (__builtin_expect (prevword == oldword, 1))                      \
> +     return oldval;                                                       \
> +      oldword = prevword;                                                 \
> +    }                                                                             \
> +}                                                                         \
> +                                                                          \
> +bool                                                                      \
> +__sync_bool_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)    \
> +{                                                                         \
> +  return __sync_val_compare_and_swap_##SIZE (ptr, oldval, newval) == oldval; \
> +}
> +
> +__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned char, 1)
> +__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned short, 2)
> +
> diff --git a/libgcc/config/nvptx/t-nvptx b/libgcc/config/nvptx/t-nvptx
> index c4d20c9..ede0bf0 100644
> --- a/libgcc/config/nvptx/t-nvptx
> +++ b/libgcc/config/nvptx/t-nvptx
> @@ -1,5 +1,6 @@
>  LIB2ADD=$(srcdir)/config/nvptx/reduction.c \
> -     $(srcdir)/config/nvptx/mgomp.c
> +     $(srcdir)/config/nvptx/mgomp.c \
> +     $(srcdir)/config/nvptx/atomic.c
>
>  LIB2ADDEH=
>  LIB2FUNCS_EXCLUDE=__main
> diff --git a/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
> new file mode 100644
> index 0000000..951e522
> --- /dev/null
> +++ b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
> @@ -0,0 +1,46 @@
> +/* { dg-do run } */
> +
> +#include <stdlib.h>
> +
> +#define N 512
> +
> +#define GENERATE_TEST(T)     \
> +int test_##T (void)          \
> +{                            \
> +  T a[N], res = 0;           \
> +                             \
> +  for (int i = 0; i < N; ++i)        \
> +    a[i] = i & 1;            \
> +                             \
> +_Pragma("omp target teams distribute reduction(||:res) defaultmap(tofrom:scalar)") \
> +  for (int i = 0; i < N; ++i)        \
> +    res = res || a[i];               \
> +                             \
> +  /* res should be non-zero.  */\
> +  if (!res)                  \
> +    return 1;                        \
> +                             \
> +_Pragma("omp target teams distribute reduction(&&:res) defaultmap(tofrom:scalar)") \
> +  for (int i = 0; i < N; ++i)        \
> +    res = res && a[i];               \
> +                             \
> +  /* res should be zero.  */ \
> +  return res;                        \
> +}
> +
> +GENERATE_TEST(char)
> +GENERATE_TEST(short)
> +GENERATE_TEST(int)
> +GENERATE_TEST(long)
> +
> +int main(void)
> +{
> +  if (test_char ())
> +    abort ();
> +  if (test_short ())
> +    abort ();
> +  if (test_int ())
> +    abort ();
> +  if (test_long ())
> +    abort ();
> +}
-----------------
Mentor Graphics (Deutschland) GmbH, Arnulfstraße 201, 80634 München / Germany
Registergericht München HRB 106955, Geschäftsführer: Thomas Heurung, Alexander Walter
Jakub Jelinek June 23, 2020, 4:51 p.m. UTC | #2
On Tue, Jun 23, 2020 at 06:44:26PM +0200, Thomas Schwinge wrote:
> On 2020-06-15T21:28:12+0100, Kwok Cheung Yeung <kcy@codesourcery.com> wrote:
> > This patch adds support on nvptx for __sync_val_compare_and_swap operations on
> > 1- and 2-byte values.
> 
> Is this a thorough review that these are the only functions missing, or
> did you just implement what you found missing for some test case you've
> been looking into?  Other architectures' similar libgcc files seem to be
> defining more of such related functions.
> It seems a bit unfortunate to have such a thing outlined in a separate
> function, given we're talking about performance-critical code here?  Even
> more so for GCN, where there's no JIT compiler that can inline it later,
> as it's the case for nvptx?

I think this should really be handled by the backend inline, like many other
targets do it when they only support 32-bit+ and not 8/16-bit atomics.
See e.g. sparc backend.

	Jakub
Kwok Cheung Yeung June 24, 2020, 11:13 a.m. UTC | #3
On 23/06/2020 5:44 pm, Thomas Schwinge wrote:
> Hi!
> 
> On 2020-06-15T21:28:12+0100, Kwok Cheung Yeung <kcy@codesourcery.com> wrote:
>> This patch adds support on nvptx for __sync_val_compare_and_swap operations on
>> 1- and 2-byte values.
> 
> Is this a thorough review that these are the only functions missing, or
> did you just implement what you found missing for some test case you've
> been looking into?  Other architectures' similar libgcc files seem to be
> defining more of such related functions.
> 

This was to fix a particular test case that used logical reduction operators on 
subword values. Support for other operators (e.g. arithmetic add) would need 
additional support.

Other architectures also define atomic operations for subwords in libgcc (e.g. 
in config/arm/linux-atomic.c for Arm, config/riscv/atomic.c for RISC-V, in 
config/m68k/linux-atomic.c for Motorola 68000s etc.).

>> The implementation is a straight copy of the version for
>> AMD GCN.
> 
> (Should thus be generalized?  That can be done later, as far as I'm
> concerned -- there already seems to be quite some code duplication in
> libgcc.)
> 
> I have not verified the algorithm.
> 
> It seems a bit unfortunate to have such a thing outlined in a separate
> function, given we're talking about performance-critical code here?  Even
> more so for GCN, where there's no JIT compiler that can inline it later,
> as it's the case for nvptx?
> 
> You have verified that GCC itself shouldn't/can't synthesize such
> replacement code, inline?
> 
> The GCN/nvptx libgcc code actually seems simple enough so that GCC could
> synthesize that internally -- or is there a reason not to?  (Just
> curious.)
>

Without the additional __sync_val_compare_and_swap_* functions, the testcase 
fails with:

unresolved symbol __sync_val_compare_and_swap_2

when the accelerator compiler is run.

The algorithm is generic and fairly simple, so it is more that GCC doesn't, 
rather than shouldn't or can't. i.e. No one has implemented it as a fallback in 
lieu of native support so far.

> I see 'gcc/doc/generic.texi', "OMP_ATOMIC" state:
> 
>      The gimplifier tries
>      three alternative code generation strategies.  Whenever possible,
>      an atomic update built-in is used.  If that fails, a
>      compare-and-swap loop is attempted.  If that also fails, a
>      regular critical section around the expression is used.
> 
> ..., and I see 'gcc/omp-expand.c:expand_omp_atomic_pipeline' synthesize
> that "compare-and-swap loop" code, which looks vaguely similar to your
> libgcc implementation.
> 
 > ..., and 'gcc/optabs.c:expand_compare_and_swap_loop' etc. also look
 > similar/related?

These create loops that repeatedly do the compare-and-swap operation until it 
succeeds. They cannot work to express a smaller compare-and-swap in terms of a 
larger one since they lack the necessary shifts and masking.

The exit condition is also different - since we are implementing 
compare-and-swap, the loop can exit even if the swap fails. The loop exits if 
the subword read from memory is different from the expected value (i.e. the 
compare-and-swap fails) or if the word containing the subword is equal to the 
expected value (i.e. it succeeds). The loop is there to cope with the case where 
the compare-and-swap fails due to part of the larger word changing that is not 
part of the subword that we are interested in.

> ..., or maybe even 'gcc/builtins.c:expand_builtin_compare_and_swap'
> etc. could be doing such things?
> 

This delegates to expand_atomic_compare_and_swap (in optabs.c), which would be 
the logical place to synthesize the sub-word loop. All it does ATM is look for 
atomic_compare_and_swap op handlers, then for sync_compare_and_swap handlers, 
then library functions for __sync_val_compare_and_swap.

Kwok
Tom de Vries June 30, 2020, 2:37 p.m. UTC | #4
On 6/23/20 6:44 PM, Thomas Schwinge wrote:
> Per the PTX 3.1 manual that I looked into, I see for CAS it supports:
> '.b32', '.b64'.  We've got:
> 
>     $ build-gcc-offload-nvptx-none/gcc/xgcc -Bbuild-gcc-offload-nvptx-none/gcc -dM -E -x c /dev/null | grep -i compare.and.swap
>     #define __GCC_HAVE_SYNC_COMPARE_AND_SWAP_4 1
>     #define __GCC_HAVE_SYNC_COMPARE_AND_SWAP_8 1
> 
> ..., so that would match the PTX 3.1 manual.  GCC also seems to know
> about 16-byte ("'.b128'", which doesn't exist in PTX).  A quick run of
> your testcase with 'GENERATE_TEST(__int128)' seems to just work, but I
> haven't verified why/how.

In this case, we hit the critical section strategy and use
expand_omp_atomic_mutex.

Thanks,
- Tom
Kwok Cheung Yeung June 30, 2020, 4:35 p.m. UTC | #5
On 23/06/2020 5:51 pm, Jakub Jelinek wrote:
> On Tue, Jun 23, 2020 at 06:44:26PM +0200, Thomas Schwinge wrote:
>> On 2020-06-15T21:28:12+0100, Kwok Cheung Yeung <kcy@codesourcery.com> wrote:
>>> This patch adds support on nvptx for __sync_val_compare_and_swap operations on
>>> 1- and 2-byte values.
>>
>> Is this a thorough review that these are the only functions missing, or
>> did you just implement what you found missing for some test case you've
>> been looking into?  Other architectures' similar libgcc files seem to be
>> defining more of such related functions.
>> It seems a bit unfortunate to have such a thing outlined in a separate
>> function, given we're talking about performance-critical code here?  Even
>> more so for GCN, where there's no JIT compiler that can inline it later,
>> as it's the case for nvptx?
> 
> I think this should really be handled by the backend inline, like many other
> targets do it when they only support 32-bit+ and not 8/16-bit atomics.
> See e.g. sparc backend.

I can see both approaches being used - e.g. Sparc, Alpha, and RS6000 expand the 
subword atomics into loops in the backend. However, there are various 
linux-atomic.c files in the subdirectories of libgcc/config/ (e.g. for Arm, 
PA-RISC, m68k etc.), which use a system-call to do the 32-bit compare-and-swap, 
then express smaller operations in terms of that. TilePro, AMDGCN and RISC-V 
have atomic.c files that do the subword compare-and-swap in terms of a larger 
native compare-and-swap.

I don't know - is it worth doing this in the backend on this architecture? One 
thing to keep in mind is that nvptx is a virtual instruction set, so the JIT 
compiler would likely inline the libgcc library calls anyway.

Thanks

Kwok
Tom de Vries July 1, 2020, 2:28 p.m. UTC | #6
On 6/15/20 10:28 PM, Kwok Cheung Yeung wrote:
> Hello
> 
> This patch adds support on nvptx for __sync_val_compare_and_swap
> operations on 1- and 2-byte values. The implementation is a straight
> copy of the version for AMD GCN.

Actually it also adds support for __sync_bool_compare_and_swap, be sure
to mention this in the commit log.

> I have added a new libgomp test that exercises the new operation. I have
> also verified that the new code does not cause any regressions on the
> nvptx offloading tests, and that the new test passes with both nvptx and
> amdgcn as offload targets.
> 

AFAICT, that excercises __sync_val_compare_and_swap, but not
__sync_bool_compare_and_swap.

You've covered offloading, but now consider the nvptx standalone target.

I was surprised to find only one generic test-case exercising char/short
__sync_val_compare_and_swap:
...
$ find gcc/testsuite/ -type f | xargs grep -l sync_char_short | xargs
grep sync_val
gcc/testsuite/gcc.dg/pr49696.c:  __sync_val_compare_and_swap (x, 1, 0);
...
and that's not a run test.

So, I think gcc needs a copy of (some of) the
gcc/testsuite/gcc.dg/ia64-sync-*.c tests for effective target
sync_char_short.

However, since this patch only adds partial support, we cannot enable
sync_char_short for nvptx yet.  So, if you stick to partial support, you
should add a char/short copy of ia64-sync-3.c to gcc.target/nvptx (which
ideally could be an include of a generic test-case that is active for
sync_char_short only, with mention that it can be removed once
sync_char_short is enabled for nvptx).

> Okay for master and OG10?
> 

I looked at the implementation, and it looks ok to me, though I think we
need to make explicit in a comment what the assumptions are:
- that we have read and write access to the entire word, and
- that the word is not volatile.

As for the discussion about libgcc vs backend implementation, I think
it's good to have both.  In particular, a backend implementation might
be able to take advantage of the address space and issue an atom.shared.cas.

But since the current patch fixes something, I don't want to hold it
until a backend implementation is done.

So, config/nvptx part okay for master, provided:
- nvptx test-case added
- nvptx standalone target testing done
- assumption comment added to implementation.
This bit can be committed as a separate patch, without the oacc test-case.

As for the oacc test-case, you could add the __int128 bit, perhaps along
the lines of how things are done in
libgomp/testsuite/libgomp.c++/target-8.C ?

[ FWIW, it would be nice if the choice between critical section and
atomic builtin was postponed till after the host/offload compiler split.
 Then the nvptx compiler would just fall back on the critical section,
and the test-case would have worked out of the box, and there would be
no need for the target to pretend to support unsupported atomic operators.
Conversely, we could upgrade the omp variables from char/short to int
somehow to make sure the assumptions for using non-native subword
__sync_val_compare_and_swap are met. ]

Thanks,
- Tom

> Kwok
> 
> 
> nvptx_subword.patch
> 
> commit 7c3a9c23ba9f5b8fe953aa5492ae75617f2444a3
> Author: Kwok Cheung Yeung <kcy@codesourcery.com>
> Date:   Mon Jun 15 12:34:55 2020 -0700
> 
>     nvptx: Add support for subword compare-and-swap
>     
>     2020-06-15  Kwok Cheung Yeung  <kcy@codesourcery.com>
>     
>     	libgcc/
>     	* config/nvptx/atomic.c: New.
>     	* config/nvptx/t-nvptx (LIB2ADD): Add atomic.c.
>     
>     	libgomp/
>     	* testsuite/libgomp.c-c++-common/reduction-16.c: New.
> 
> diff --git a/libgcc/config/nvptx/atomic.c b/libgcc/config/nvptx/atomic.c
> new file mode 100644
> index 0000000..4becbd2
> --- /dev/null
> +++ b/libgcc/config/nvptx/atomic.c
> @@ -0,0 +1,59 @@
> +/* NVPTX atomic operations
> +   Copyright (C) 2020 Free Software Foundation, Inc.
> +   Contributed by Mentor Graphics.
> +
> +   This file is free software; you can redistribute it and/or modify it
> +   under the terms of the GNU General Public License as published by the
> +   Free Software Foundation; either version 3, or (at your option) any
> +   later version.
> +
> +   This file is distributed in the hope that it will be useful, but
> +   WITHOUT ANY WARRANTY; without even the implied warranty of
> +   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
> +   General Public License for more details.
> +
> +   Under Section 7 of GPL version 3, you are granted additional
> +   permissions described in the GCC Runtime Library Exception, version
> +   3.1, as published by the Free Software Foundation.
> +
> +   You should have received a copy of the GNU General Public License and
> +   a copy of the GCC Runtime Library Exception along with this program;
> +   see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
> +   <http://www.gnu.org/licenses/>.  */
> +
> +#include <stdbool.h>
> +
> +#define __SYNC_SUBWORD_COMPARE_AND_SWAP(TYPE, SIZE)			     \
> +									     \
> +TYPE									     \
> +__sync_val_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)     \
> +{									     \
> +  unsigned int *wordptr = (unsigned int *)((__UINTPTR_TYPE__ ) ptr & ~3UL);  \
> +  int shift = ((__UINTPTR_TYPE__ ) ptr & 3UL) * 8;			     \
> +  unsigned int valmask = (1 << (SIZE * 8)) - 1;				     \
> +  unsigned int wordmask = ~(valmask << shift);				     \
> +  unsigned int oldword = *wordptr;					     \
> +  for (;;)								     \
> +    {									     \
> +      TYPE prevval = (oldword >> shift) & valmask;			     \
> +      if (__builtin_expect (prevval != oldval, 0))			     \
> +	return prevval;							     \
> +      unsigned int newword = oldword & wordmask;			     \
> +      newword |= ((unsigned int) newval) << shift;			     \
> +      unsigned int prevword						     \
> +	  = __sync_val_compare_and_swap_4 (wordptr, oldword, newword);	     \
> +      if (__builtin_expect (prevword == oldword, 1))			     \
> +	return oldval;							     \
> +      oldword = prevword;						     \
> +    }									     \
> +}									     \
> +									     \
> +bool									     \
> +__sync_bool_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)    \
> +{									     \
> +  return __sync_val_compare_and_swap_##SIZE (ptr, oldval, newval) == oldval; \
> +}
> +
> +__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned char, 1)
> +__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned short, 2)
> +
> diff --git a/libgcc/config/nvptx/t-nvptx b/libgcc/config/nvptx/t-nvptx
> index c4d20c9..ede0bf0 100644
> --- a/libgcc/config/nvptx/t-nvptx
> +++ b/libgcc/config/nvptx/t-nvptx
> @@ -1,5 +1,6 @@
>  LIB2ADD=$(srcdir)/config/nvptx/reduction.c \
> -	$(srcdir)/config/nvptx/mgomp.c
> +	$(srcdir)/config/nvptx/mgomp.c \
> +	$(srcdir)/config/nvptx/atomic.c
>  
>  LIB2ADDEH=
>  LIB2FUNCS_EXCLUDE=__main
> diff --git a/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
> new file mode 100644
> index 0000000..951e522
> --- /dev/null
> +++ b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
> @@ -0,0 +1,46 @@
> +/* { dg-do run } */
> +
> +#include <stdlib.h>
> +
> +#define N 512
> +
> +#define GENERATE_TEST(T)	\
> +int test_##T (void)		\
> +{				\
> +  T a[N], res = 0;		\
> +				\
> +  for (int i = 0; i < N; ++i)	\
> +    a[i] = i & 1;		\
> +				\
> +_Pragma("omp target teams distribute reduction(||:res) defaultmap(tofrom:scalar)") \
> +  for (int i = 0; i < N; ++i)	\
> +    res = res || a[i];		\
> +				\
> +  /* res should be non-zero.  */\
> +  if (!res)			\
> +    return 1;			\
> +				\
> +_Pragma("omp target teams distribute reduction(&&:res) defaultmap(tofrom:scalar)") \
> +  for (int i = 0; i < N; ++i)	\
> +    res = res && a[i];		\
> +				\
> +  /* res should be zero.  */	\
> +  return res;			\
> +}
> +
> +GENERATE_TEST(char)
> +GENERATE_TEST(short)
> +GENERATE_TEST(int)
> +GENERATE_TEST(long)
> +
> +int main(void)
> +{
> +  if (test_char ())
> +    abort ();
> +  if (test_short ())
> +    abort ();
> +  if (test_int ())
> +    abort ();
> +  if (test_long ())
> +    abort ();
> +}
>
Kwok Cheung Yeung July 15, 2020, 7:08 p.m. UTC | #7
On 01/07/2020 3:28 pm, Tom de Vries wrote:
> I looked at the implementation, and it looks ok to me, though I think we
> need to make explicit in a comment what the assumptions are:
> - that we have read and write access to the entire word, and

Is there a situation where an 8/16-bit portion of memory is R/W but the 32-bit 
word containing it is not on this architecture? Something like memory-mapped I/O 
perhaps?

> - that the word is not volatile.

I don't think that non-volatility matters in this case - indeed, the whole point 
of using an atomic primitive is that the memory being accessed can change at any 
time :-). There is an initial read of the word before the loop, then one read 
per iteration of the loop using __sync_val_compare_and_swap_4 (which by 
definition will always access the memory). If any part of the initially read 
value changes by the time it gets to __sync_val_compare_and_swap_4, then the 
compare will fail and the loop will continue onto the next iteration.

Thanks

Kwok
Kwok Cheung Yeung July 20, 2020, 1:19 p.m. UTC | #8
On 01/07/2020 3:28 pm, Tom de Vries wrote:
> So, I think gcc needs a copy of (some of) the
> gcc/testsuite/gcc.dg/ia64-sync-*.c tests for effective target
> sync_char_short.
> 
> However, since this patch only adds partial support, we cannot enable
> sync_char_short for nvptx yet.  So, if you stick to partial support, you
> should add a char/short copy of ia64-sync-3.c to gcc.target/nvptx (which
> ideally could be an include of a generic test-case that is active for
> sync_char_short only, with mention that it can be removed once
> sync_char_short is enabled for nvptx).
> 

I have added gcc.target/nvptx/sync.c, which is a version of ia64-sync-3.c 
extended to test chars and shorts too. I kept the original int and long tests 
because sync_int_long isn't indicated as being supported on nvptx either.

> I looked at the implementation, and it looks ok to me, though I think we
> need to make explicit in a comment what the assumptions are:
> - that we have read and write access to the entire word, and
> - that the word is not volatile.
> 

I've added some extra comments in the implementation. Like I said previously, 
the loop accounts for the larger word being volatile.

> As for the oacc test-case, you could add the __int128 bit, perhaps along
> the lines of how things are done in
> libgomp/testsuite/libgomp.c++/target-8.C ?
> 

I've added a extra test for __int128 types in my libgomp testcase that runs if 
128-bit types are supported.

I've tested that there are no regressions with the patch on standalone nvptx, 
and that the new reduction-16.c testcase passes with both nvptx and AMD GCN 
offloading.

Is this version okay for master and og10?

Thanks

Kwok
commit 4661232905d55a4bc1354cb717b2e5d950d215af
Author: Kwok Cheung Yeung <kcy@codesourcery.com>
Date:   Thu Jul 16 12:00:24 2020 -0700

    nvptx: Add support for subword compare-and-swap
    
    This adds support for __sync_val_compare_and_swap and
    __sync_bool_compare_and_swap for 1-byte and 2-byte long
    values, which are not natively supported on nvptx.
    
    2020-07-16  Kwok Cheung Yeung  <kcy@codesourcery.com>
    
    	libgcc/
    	* config/nvptx/atomic.c: New.
    	* config/nvptx/t-nvptx (LIB2ADD): Add atomic.c.
    
    	gcc/testsuite/
    	* gcc.target/nvptx/sync.c: New.
    
    	libgomp/
    	* testsuite/libgomp.c-c++-common/reduction-16.c: New.

diff --git a/gcc/testsuite/gcc.target/nvptx/sync.c b/gcc/testsuite/gcc.target/nvptx/sync.c
new file mode 100644
index 0000000..a573824
--- /dev/null
+++ b/gcc/testsuite/gcc.target/nvptx/sync.c
@@ -0,0 +1,143 @@
+/* { dg-do run } */
+
+/* Test basic functionality of the intrinsics.  */
+
+/* This is a copy of gcc.dg/ia64-sync-2.c, extended to test 8-bit and 16-bit
+   values as well.  */
+
+/* Ideally this test should require sync_char_short and sync_int_long, but we
+   only support a subset at the moment.  */
+
+__extension__ typedef __SIZE_TYPE__ size_t;
+
+extern void abort (void);
+extern void *memcpy (void *, const void *, size_t);
+extern int memcmp (const void *, const void *, size_t);
+
+static char AC[4];
+static char init_qi[4] = { -30,-30,-50,-50 };
+static char test_qi[4] = { -115,-115,25,25 };
+
+static void
+do_qi (void)
+{
+  if (__sync_val_compare_and_swap(AC+0, -30, -115) != -30)
+    abort ();
+  if (__sync_val_compare_and_swap(AC+0, -30, -115) != -115)
+    abort ();
+  if (__sync_bool_compare_and_swap(AC+1, -30, -115) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AC+1, -30, -115) != 0)
+    abort ();
+
+  if (__sync_val_compare_and_swap(AC+2, AC[2], 25) != -50)
+    abort ();
+  if (__sync_val_compare_and_swap(AC+2, AC[2], 25) != 25)
+    abort ();
+  if (__sync_bool_compare_and_swap(AC+3, AC[3], 25) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AC+3, AC[3], 25) != 1)
+    abort ();
+}
+
+static short AS[4];
+static short init_hi[4] = { -30,-30,-50,-50 };
+static short test_hi[4] = { -115,-115,25,25 };
+
+static void
+do_hi (void)
+{
+  if (__sync_val_compare_and_swap(AS+0, -30, -115) != -30)
+    abort ();
+  if (__sync_val_compare_and_swap(AS+0, -30, -115) != -115)
+    abort ();
+  if (__sync_bool_compare_and_swap(AS+1, -30, -115) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AS+1, -30, -115) != 0)
+    abort ();
+
+  if (__sync_val_compare_and_swap(AS+2, AS[2], 25) != -50)
+    abort ();
+  if (__sync_val_compare_and_swap(AS+2, AS[2], 25) != 25)
+    abort ();
+  if (__sync_bool_compare_and_swap(AS+3, AS[3], 25) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AS+3, AS[3], 25) != 1)
+    abort ();
+}
+
+static int AI[4];
+static int init_si[4] = { -30,-30,-50,-50 };
+static int test_si[4] = { -115,-115,25,25 };
+
+static void
+do_si (void)
+{
+  if (__sync_val_compare_and_swap(AI+0, -30, -115) != -30)
+    abort ();
+  if (__sync_val_compare_and_swap(AI+0, -30, -115) != -115)
+    abort ();
+  if (__sync_bool_compare_and_swap(AI+1, -30, -115) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AI+1, -30, -115) != 0)
+    abort ();
+
+  if (__sync_val_compare_and_swap(AI+2, AI[2], 25) != -50)
+    abort ();
+  if (__sync_val_compare_and_swap(AI+2, AI[2], 25) != 25)
+    abort ();
+  if (__sync_bool_compare_and_swap(AI+3, AI[3], 25) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AI+3, AI[3], 25) != 1)
+    abort ();
+}
+
+static long AL[4];
+static long init_di[4] = { -30,-30,-50,-50 };
+static long test_di[4] = { -115,-115,25,25 };
+
+static void
+do_di (void)
+{
+  if (__sync_val_compare_and_swap(AL+0, -30, -115) != -30)
+    abort ();
+  if (__sync_val_compare_and_swap(AL+0, -30, -115) != -115)
+    abort ();
+  if (__sync_bool_compare_and_swap(AL+1, -30, -115) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AL+1, -30, -115) != 0)
+    abort ();
+
+  if (__sync_val_compare_and_swap(AL+2, AL[2], 25) != -50)
+    abort ();
+  if (__sync_val_compare_and_swap(AL+2, AL[2], 25) != 25)
+    abort ();
+  if (__sync_bool_compare_and_swap(AL+3, AL[3], 25) != 1)
+    abort ();
+  if (__sync_bool_compare_and_swap(AL+3, AL[3], 25) != 1)
+    abort ();
+}
+
+int main()
+{
+  memcpy(AC, init_qi, sizeof(init_qi));
+  memcpy(AS, init_hi, sizeof(init_hi));
+  memcpy(AI, init_si, sizeof(init_si));
+  memcpy(AL, init_di, sizeof(init_di));
+
+  do_qi ();
+  do_hi ();
+  do_si ();
+  do_di ();
+
+  if (memcmp (AC, test_qi, sizeof(test_qi)))
+    abort ();
+  if (memcmp (AS, test_hi, sizeof(test_hi)))
+    abort ();
+  if (memcmp (AI, test_si, sizeof(test_si)))
+    abort ();
+  if (memcmp (AL, test_di, sizeof(test_di)))
+    abort ();
+
+  return 0;
+}
diff --git a/libgcc/config/nvptx/atomic.c b/libgcc/config/nvptx/atomic.c
new file mode 100644
index 0000000..25a34fb
--- /dev/null
+++ b/libgcc/config/nvptx/atomic.c
@@ -0,0 +1,70 @@
+/* NVPTX atomic operations
+   Copyright (C) 2020 Free Software Foundation, Inc.
+   Contributed by Mentor Graphics.
+
+   This file is free software; you can redistribute it and/or modify it
+   under the terms of the GNU General Public License as published by the
+   Free Software Foundation; either version 3, or (at your option) any
+   later version.
+
+   This file is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   General Public License for more details.
+
+   Under Section 7 of GPL version 3, you are granted additional
+   permissions described in the GCC Runtime Library Exception, version
+   3.1, as published by the Free Software Foundation.
+
+   You should have received a copy of the GNU General Public License and
+   a copy of the GCC Runtime Library Exception along with this program;
+   see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
+   <http://www.gnu.org/licenses/>.  */
+
+#include <stdbool.h>
+
+/* Implement __sync_val_compare_and_swap and __sync_bool_compare_and_swap
+   for 1 and 2-byte values (which are not natively supported) in terms of
+   __sync_val_compare_and_swap for 4-byte values (which is supported).
+   This assumes that the contents of the word surrounding the subword
+   value that we are interested in are accessible as well (which should
+   normally be the case).  */
+
+#define __SYNC_SUBWORD_COMPARE_AND_SWAP(TYPE, SIZE)			     \
+									     \
+TYPE									     \
+__sync_val_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)     \
+{									     \
+  unsigned int *wordptr = (unsigned int *)((__UINTPTR_TYPE__ ) ptr & ~3UL);  \
+  int shift = ((__UINTPTR_TYPE__ ) ptr & 3UL) * 8;			     \
+  unsigned int valmask = (1 << (SIZE * 8)) - 1;				     \
+  unsigned int wordmask = ~(valmask << shift);				     \
+  unsigned int oldword = *wordptr;					     \
+  for (;;)								     \
+    {									     \
+      TYPE prevval = (oldword >> shift) & valmask;			     \
+      /* Exit if the subword value previously read from memory is not */     \
+      /* equal to the expected value OLDVAL.  */			     \
+      if (__builtin_expect (prevval != oldval, 0))			     \
+	return prevval;							     \
+      unsigned int newword = oldword & wordmask;			     \
+      newword |= ((unsigned int) newval) << shift;			     \
+      unsigned int prevword						     \
+	  = __sync_val_compare_and_swap_4 (wordptr, oldword, newword);	     \
+      /* Exit only if the compare-and-swap succeeds on the whole word */     \
+      /* (i.e. the contents of *WORDPTR have not changed since the last */   \
+      /* memory read).  */						     \
+      if (__builtin_expect (prevword == oldword, 1))			     \
+	return oldval;							     \
+      oldword = prevword;						     \
+    }									     \
+}									     \
+									     \
+bool									     \
+__sync_bool_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)    \
+{									     \
+  return __sync_val_compare_and_swap_##SIZE (ptr, oldval, newval) == oldval; \
+}
+
+__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned char, 1)
+__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned short, 2)
diff --git a/libgcc/config/nvptx/t-nvptx b/libgcc/config/nvptx/t-nvptx
index c4d20c9..ede0bf0 100644
--- a/libgcc/config/nvptx/t-nvptx
+++ b/libgcc/config/nvptx/t-nvptx
@@ -1,5 +1,6 @@
 LIB2ADD=$(srcdir)/config/nvptx/reduction.c \
-	$(srcdir)/config/nvptx/mgomp.c
+	$(srcdir)/config/nvptx/mgomp.c \
+	$(srcdir)/config/nvptx/atomic.c
 
 LIB2ADDEH=
 LIB2FUNCS_EXCLUDE=__main
diff --git a/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
new file mode 100644
index 0000000..d0e82b0
--- /dev/null
+++ b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
@@ -0,0 +1,53 @@
+/* { dg-do run } */
+
+#include <stdlib.h>
+
+#define N 512
+
+#define GENERATE_TEST(T)	\
+int test_##T (void)		\
+{				\
+  T a[N], res = 0;		\
+				\
+  for (int i = 0; i < N; ++i)	\
+    a[i] = i & 1;		\
+				\
+_Pragma("omp target teams distribute reduction(||:res) defaultmap(tofrom:scalar)") \
+  for (int i = 0; i < N; ++i)	\
+    res = res || a[i];		\
+				\
+  /* res should be non-zero.  */\
+  if (!res)			\
+    return 1;			\
+				\
+_Pragma("omp target teams distribute reduction(&&:res) defaultmap(tofrom:scalar)") \
+  for (int i = 0; i < N; ++i)	\
+    res = res && a[i];		\
+				\
+  /* res should be zero.  */	\
+  return res;			\
+}
+
+GENERATE_TEST(char)
+GENERATE_TEST(short)
+GENERATE_TEST(int)
+GENERATE_TEST(long)
+#ifdef __SIZEOF_INT128__
+GENERATE_TEST(__int128)
+#endif
+
+int main(void)
+{
+  if (test_char ())
+    abort ();
+  if (test_short ())
+    abort ();
+  if (test_int ())
+    abort ();
+  if (test_long ())
+    abort ();
+#ifdef __SIZEOF_INT128__
+  if (test___int128 ())
+    abort ();
+#endif
+}
Kwok Cheung Yeung Aug. 4, 2020, 2:56 p.m. UTC | #9
Hello

I posted a revised patchset about two weeks ago at:

https://gcc.gnu.org/pipermail/gcc-patches/2020-July/550291.html

Are you able to take a look at it?

Thanks

Kwok
Tom de Vries Aug. 13, 2020, 9:27 a.m. UTC | #10
On 7/20/20 3:19 PM, Kwok Cheung Yeung wrote:
> On 01/07/2020 3:28 pm, Tom de Vries wrote:
>> So, I think gcc needs a copy of (some of) the
>> gcc/testsuite/gcc.dg/ia64-sync-*.c tests for effective target
>> sync_char_short.
>>
>> However, since this patch only adds partial support, we cannot enable
>> sync_char_short for nvptx yet.  So, if you stick to partial support, you
>> should add a char/short copy of ia64-sync-3.c to gcc.target/nvptx (which
>> ideally could be an include of a generic test-case that is active for
>> sync_char_short only, with mention that it can be removed once
>> sync_char_short is enabled for nvptx).
>>
> 
> I have added gcc.target/nvptx/sync.c, which is a version of
> ia64-sync-3.c extended to test chars and shorts too.

I've:
- added the short/char part of that as gcc.dg/ia64-async-5.c
- included the sync_int_long ones of gcc.dg/ia64-async-* in
  gcc.target/nvptx
- did the same for gcc.dg/ia64-async-5.c as part of this patch

> I kept the original
> int and long tests because sync_int_long isn't indicated as being
> supported on nvptx either.
> 

Yep, I proposed a patch to enable that:
https://gcc.gnu.org/pipermail/gcc-patches/2020-August/551842.html .

>> I looked at the implementation, and it looks ok to me, though I think we
>> need to make explicit in a comment what the assumptions are:
>> - that we have read and write access to the entire word, and
>> - that the word is not volatile.
>>
> 
> I've added some extra comments in the implementation. Like I said
> previously, the loop accounts for the larger word being volatile.
> 

Right, I known what that loop is intending to do. The loop though may
manifest worst-case as a hang, so I've mentioned that in the comment.

>> As for the oacc test-case, you could add the __int128 bit, perhaps along
>> the lines of how things are done in
>> libgomp/testsuite/libgomp.c++/target-8.C ?
>>
> 
> I've added a extra test for __int128 types in my libgomp testcase that
> runs if 128-bit types are supported.
> 
> I've tested that there are no regressions with the patch on standalone
> nvptx, and that the new reduction-16.c testcase passes with both nvptx
> and AMD GCN offloading.
> 
> Is this version okay for master and og10?
> 

Pushed as attached to master.

Thanks,
- Tom
diff mbox series

Patch

diff --git a/libgcc/config/nvptx/atomic.c b/libgcc/config/nvptx/atomic.c
new file mode 100644
index 0000000..4becbd2
--- /dev/null
+++ b/libgcc/config/nvptx/atomic.c
@@ -0,0 +1,59 @@ 
+/* NVPTX atomic operations
+   Copyright (C) 2020 Free Software Foundation, Inc.
+   Contributed by Mentor Graphics.
+
+   This file is free software; you can redistribute it and/or modify it
+   under the terms of the GNU General Public License as published by the
+   Free Software Foundation; either version 3, or (at your option) any
+   later version.
+
+   This file is distributed in the hope that it will be useful, but
+   WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   General Public License for more details.
+
+   Under Section 7 of GPL version 3, you are granted additional
+   permissions described in the GCC Runtime Library Exception, version
+   3.1, as published by the Free Software Foundation.
+
+   You should have received a copy of the GNU General Public License and
+   a copy of the GCC Runtime Library Exception along with this program;
+   see the files COPYING3 and COPYING.RUNTIME respectively.  If not, see
+   <http://www.gnu.org/licenses/>.  */
+
+#include <stdbool.h>
+
+#define __SYNC_SUBWORD_COMPARE_AND_SWAP(TYPE, SIZE)			     \
+									     \
+TYPE									     \
+__sync_val_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)     \
+{									     \
+  unsigned int *wordptr = (unsigned int *)((__UINTPTR_TYPE__ ) ptr & ~3UL);  \
+  int shift = ((__UINTPTR_TYPE__ ) ptr & 3UL) * 8;			     \
+  unsigned int valmask = (1 << (SIZE * 8)) - 1;				     \
+  unsigned int wordmask = ~(valmask << shift);				     \
+  unsigned int oldword = *wordptr;					     \
+  for (;;)								     \
+    {									     \
+      TYPE prevval = (oldword >> shift) & valmask;			     \
+      if (__builtin_expect (prevval != oldval, 0))			     \
+	return prevval;							     \
+      unsigned int newword = oldword & wordmask;			     \
+      newword |= ((unsigned int) newval) << shift;			     \
+      unsigned int prevword						     \
+	  = __sync_val_compare_and_swap_4 (wordptr, oldword, newword);	     \
+      if (__builtin_expect (prevword == oldword, 1))			     \
+	return oldval;							     \
+      oldword = prevword;						     \
+    }									     \
+}									     \
+									     \
+bool									     \
+__sync_bool_compare_and_swap_##SIZE (TYPE *ptr, TYPE oldval, TYPE newval)    \
+{									     \
+  return __sync_val_compare_and_swap_##SIZE (ptr, oldval, newval) == oldval; \
+}
+
+__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned char, 1)
+__SYNC_SUBWORD_COMPARE_AND_SWAP (unsigned short, 2)
+
diff --git a/libgcc/config/nvptx/t-nvptx b/libgcc/config/nvptx/t-nvptx
index c4d20c9..ede0bf0 100644
--- a/libgcc/config/nvptx/t-nvptx
+++ b/libgcc/config/nvptx/t-nvptx
@@ -1,5 +1,6 @@ 
 LIB2ADD=$(srcdir)/config/nvptx/reduction.c \
-	$(srcdir)/config/nvptx/mgomp.c
+	$(srcdir)/config/nvptx/mgomp.c \
+	$(srcdir)/config/nvptx/atomic.c
 
 LIB2ADDEH=
 LIB2FUNCS_EXCLUDE=__main
diff --git a/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
new file mode 100644
index 0000000..951e522
--- /dev/null
+++ b/libgomp/testsuite/libgomp.c-c++-common/reduction-16.c
@@ -0,0 +1,46 @@ 
+/* { dg-do run } */
+
+#include <stdlib.h>
+
+#define N 512
+
+#define GENERATE_TEST(T)	\
+int test_##T (void)		\
+{				\
+  T a[N], res = 0;		\
+				\
+  for (int i = 0; i < N; ++i)	\
+    a[i] = i & 1;		\
+				\
+_Pragma("omp target teams distribute reduction(||:res) defaultmap(tofrom:scalar)") \
+  for (int i = 0; i < N; ++i)	\
+    res = res || a[i];		\
+				\
+  /* res should be non-zero.  */\
+  if (!res)			\
+    return 1;			\
+				\
+_Pragma("omp target teams distribute reduction(&&:res) defaultmap(tofrom:scalar)") \
+  for (int i = 0; i < N; ++i)	\
+    res = res && a[i];		\
+				\
+  /* res should be zero.  */	\
+  return res;			\
+}
+
+GENERATE_TEST(char)
+GENERATE_TEST(short)
+GENERATE_TEST(int)
+GENERATE_TEST(long)
+
+int main(void)
+{
+  if (test_char ())
+    abort ();
+  if (test_short ())
+    abort ();
+  if (test_int ())
+    abort ();
+  if (test_long ())
+    abort ();
+}