gcc/testsuite/g++.dg/pr80481.C

   1 // { dg-do compile { target { i?86-*-* x86_64-*-* }  && { ! *-*-solaris* } } }
   2 // -fopenmp implies -pthread
   3 // { dg-require-effective-target pthread }
   4 // { dg-options "-Ofast -funroll-loops -fopenmp -march=skylake-avx512" }
   5 // Disabling epilogues until we find a better way to deal with scans.
   6 // { dg-additional-options "--param vect-epilogues-nomask=0" }
   7
   8
   9 #include <math.h>
  10
  11 #include <xmmintrin.h>
  12
  13 #define max(a, b)   ( (a) > (b) ? (a) : (b) )
  14
  15 struct Sdata {
  16   float w;
  17   float s;
  18   float r;
  19   float t;
  20   float v;
  21 };
  22  extern int N1, N2, N3;
  23
  24 #define func(p, up, down) ((p)*(up) + (1.0f-(p)) * (down))
  25
  26 void foo (Sdata *in, int idx, float *out)
  27 {
  28   float* y1 = (float*)_mm_malloc(sizeof(float) * N1,16);
  29   float* y2  = (float*)_mm_malloc(sizeof(float) * N1,16);
  30   float* y3  = (float*)_mm_malloc(sizeof(float) * N1,16);
  31   float* y4  = (float*)_mm_malloc(sizeof(float) * N1,16);
  32
  33   for (int k = idx; k < idx + N3; k++) {
  34     float x1 = in[k].r;
  35     float x2    = in[k].s;
  36     float x3      = in[k].w;
  37     float x4     = in[k].v;
  38     float x5         = in[k].t;
  39     x5 /= N2;
  40     float u = exp(x4 * sqrt(x5));
  41     float d = exp(-x4 * sqrt(x5));
  42     float a = exp(x1 * x5);
  43     float m = exp(-x1 * x5);
  44     float p = (a - d) / (u - d);
  45     y2[0] = x2;
  46     y3[0] = float(1.f);
  47     for (int i = 1; i <= N2; i++) {
  48       y2[i] = u * y2[i - 1];
  49       y3[i] = d * y3[i - 1];
  50     }
  51 #pragma omp simd
  52     for (int i = 0; i <= N2; i++) {
  53       y1[i] =
  54         max((x3 - y2[N2 - i] * y3[i]), float(0.f));
  55     }
  56     for (int i = N2 - 1; i >= 0; i--) {
  57 #pragma omp simd
  58       for (int j = 0; j <= i; j++) {
  59         y4[j] = func(p,y1[j],y1[j+1]) * m;
  60       }
  61 #pragma omp simd
  62       for (int j = 0; j <= i; j++) {
  63         float t1 = y2[i - j] * y3[j];
  64         float t2 = max(x3 - t1, float(0.f));
  65         y1[j] = max(t2, y4[j]);
  66       }
  67     }
  68     out[k] = y1[0];
  69   }
  70   _mm_free(y1);
  71   _mm_free(y2);
  72   _mm_free(y3);
  73   _mm_free(y4);
  74 }
  75
  76 // { dg-final { scan-assembler-not "vmovaps\[^\n\r]*zmm\[0-9]+,\[^\n\r]*zmm\[0-9]+" } }