gcc/testsuite/gcc.target/i386/avx512f-set-v16sf-4.c

   1 /* { dg-do run } */
   2 /* { dg-options "-O2 -mavx512f" } */
   3 /* { dg-require-effective-target avx512f } */
   4
   5 #include "avx512f-check.h"
   6
   7 static __m512
   8 __attribute__ ((noinline))
   9 foo (float x, int i)
  10 {
  11   switch (i)
  12     {
  13     case 15:
  14       return _mm512_set_ps (x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  15     case 14:
  16       return _mm512_set_ps (0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  17     case 13:
  18       return _mm512_set_ps (0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  19     case 12:
  20       return _mm512_set_ps (0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  21     case 11:
  22       return _mm512_set_ps (0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  23     case 10:
  24       return _mm512_set_ps (0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  25     case 9:
  26       return _mm512_set_ps (0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  27     case 8:
  28       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0);
  29     case 7:
  30       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0);
  31     case 6:
  32       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0);
  33     case 5:
  34       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0);
  35     case 4:
  36       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0);
  37     case 3:
  38       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0);
  39     case 2:
  40       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0);
  41     case 1:
  42       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0);
  43     case 0:
  44       return _mm512_set_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x);
  45     default:
  46       abort ();
  47     }
  48 }
  49
  50 static __m512
  51 __attribute__ ((noinline))
  52 foo_r (float x, int i)
  53 {
  54   switch (i)
  55     {
  56     case 0:
  57       return _mm512_setr_ps (x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  58     case 1:
  59       return _mm512_setr_ps (0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  60     case 2:
  61       return _mm512_setr_ps (0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  62     case 3:
  63       return _mm512_setr_ps (0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  64     case 4:
  65       return _mm512_setr_ps (0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  66     case 5:
  67       return _mm512_setr_ps (0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  68     case 6:
  69       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0, 0);
  70     case 7:
  71       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0, 0);
  72     case 8:
  73       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0, 0);
  74     case 9:
  75       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0, 0);
  76     case 10:
  77       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0, 0);
  78     case 11:
  79       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0, 0);
  80     case 12:
  81       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0, 0);
  82     case 13:
  83       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0, 0);
  84     case 14:
  85       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x, 0);
  86     case 15:
  87       return _mm512_setr_ps (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, x);
  88     default:
  89       abort ();
  90     }
  91 }
  92
  93 static void
  94 avx512f_test (void)
  95 {
  96   float e = -3.234;
  97   float v[16];
  98   union512 res;
  99   int i, j;
 100
 101   for (i = 0; i < 16; i++)
 102     {
 103       for (j = 0; j < 16; j++)
 104         v[j] = 0;
 105       v[i] = e;
 106
 107       res.x = foo (e, i);
 108
 109       if (check_union512 (res, v))
 110         abort ();
 111
 112       res.x = _mm512_setzero_ps ();
 113
 114       res.x = foo_r (e, i);
 115
 116       if (check_union512 (res, v))
 117         abort ();
 118     }
 119 }