gcc/testsuite/gcc.dg/vect/vect-simd-clone-20.c

   1 /* { dg-require-effective-target vect_simd_clones } */
   2 /* { dg-additional-options "-fopenmp-simd --param vect-epilogues-nomask=0" } */
   3 /* { dg-additional-options "-mavx" { target avx_runtime } } */
   4
   5 /* Test that simd inbranch clones work correctly.  */
   6
   7 #ifndef TYPE
   8 #define TYPE int
   9 #endif
  10
  11 /* A simple function that will be cloned.  */
  12 #pragma omp declare simd inbranch
  13 TYPE __attribute__((noinline))
  14 foo (TYPE a)
  15 {
  16   return a + 1;
  17 }
  18
  19 /* Check that "inbranch" clones are called correctly.  */
  20
  21 void __attribute__((noipa))
  22 masked (TYPE * __restrict a, TYPE * __restrict b, int size)
  23 {
  24   #pragma omp simd
  25   for (int i = 0; i < size; i++)
  26     b[i] = foo(a[i]);
  27 }
  28
  29 /* Check that "inbranch" works when there might be unrolling.  */
  30
  31 void __attribute__((noipa))
  32 masked_fixed (TYPE * __restrict a, TYPE * __restrict b)
  33 {
  34   #pragma omp simd
  35   for (int i = 0; i < 128; i++)
  36     b[i] = foo(a[i]);
  37 }
  38
  39 /* Validate the outputs.  */
  40
  41 void
  42 check_masked (TYPE *b, int size)
  43 {
  44   for (int i = 0; i < size; i++)
  45     if (b[i] != (TYPE)(i + 1))
  46       {
  47         __builtin_printf ("error at %d\n", i);
  48         __builtin_exit (1);
  49       }
  50 }
  51
  52 int
  53 main ()
  54 {
  55   TYPE a[1024];
  56   TYPE b[1024];
  57
  58   for (int i = 0; i < 1024; i++)
  59     a[i] = i;
  60
  61   masked_fixed (a, b);
  62   check_masked (b, 128);
  63
  64   /* Test various sizes to cover machines with different vectorization
  65      factors.  */
  66   for (int size = 8; size <= 1024; size *= 2)
  67     {
  68       masked (a, b, size);
  69       check_masked (b, size);
  70     }
  71
  72   /* Test sizes that might exercise the partial vector code-path.  */
  73   for (int size = 8; size <= 1024; size *= 2)
  74     {
  75       masked (a, b, size-4);
  76       check_masked (b, size-4);
  77     }
  78
  79   return 0;
  80 }
  81
  82 /* Ensure the the in-branch simd clones are used on targets that support them.  */
  83 /* { dg-final { scan-tree-dump-times {[\n\r] [^\n]* = foo\.simdclone} 2 "vect" { target { aarch64*-*-* } } } } */
  84 /* { dg-final { scan-tree-dump-times {[\n\r] [^\n]* = foo\.simdclone} 4 "vect" { target { x86_64*-*-* } } } } */
  85
  86 /* The LTO test produces two dump files and we scan the wrong one.  */
  87 /* { dg-skip-if "" { *-*-* } { "-flto" } { "" } } */