gcc/testsuite/gcc.target/riscv/rvv/autovec/ternop/ternop_run-9.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

/* { dg-do run { target { riscv_vector } } } */
/* { dg-additional-options "--param=riscv-autovec-preference=scalable -ffast-math" } */

#include "ternop-9.c"

#define TEST_LOOP(TYPE, NUM)                                                   \
  {                                                                            \
    TYPE array1_##NUM[NUM] = {};                                               \
    TYPE array2_##NUM[NUM] = {};                                               \
    TYPE array3_##NUM[NUM] = {};                                               \
    TYPE array4_##NUM[NUM] = {};                                               \
    TYPE array5_##NUM[NUM] = {};                                               \
    TYPE array6_##NUM[NUM] = {};                                               \
    TYPE array7_##NUM[NUM] = {};                                               \
    TYPE array8_##NUM[NUM] = {};                                               \
    for (int i = 0; i < NUM; ++i)                                              \
      {                                                                        \
	array1_##NUM[i] = (i & 1) + 5;                                         \
	array2_##NUM[i] = i - NUM / 3;                                         \
	array3_##NUM[i] = NUM - NUM / 3 - i;                                   \
	array6_##NUM[i] = NUM - NUM / 3 - i;                                   \
	array4_##NUM[i] = NUM - NUM / 2 + i;                                   \
	array7_##NUM[i] = NUM - NUM / 2 + i;                                   \
	array5_##NUM[i] = NUM + i * 7;                                         \
	array8_##NUM[i] = NUM + i * 7;                                         \
	asm volatile("" ::: "memory");                                         \
      }                                                                        \
    ternop_##TYPE (array3_##NUM, array4_##NUM, array5_##NUM, array1_##NUM,     \
		   array2_##NUM, NUM);                                         \
    for (int i = 0; i < NUM; i++)                                              \
      {                                                                        \
	array6_##NUM[i]                                                        \
	  = (TYPE) (array1_##NUM[i] * array2_##NUM[i] - array7_##NUM[i]);      \
	if (array3_##NUM[i] != array6_##NUM[i])                                \
	  __builtin_abort ();                                                  \
	array7_##NUM[i]                                                        \
	  = (TYPE) (array1_##NUM[i] * array6_##NUM[i] - array7_##NUM[i]);      \
	if (array4_##NUM[i] != array7_##NUM[i])                                \
	  __builtin_abort ();                                                  \
	array8_##NUM[i]                                                        \
	  = (TYPE) (array2_##NUM[i] * array7_##NUM[i] - array8_##NUM[i]);      \
	if (array5_##NUM[i] != array8_##NUM[i])                                \
	  __builtin_abort ();                                                  \
      }                                                                        \
  }

int __attribute__ ((optimize (0))) main ()
{
  TEST_LOOP (float, 7)
  TEST_LOOP (double, 7)
  TEST_LOOP (float, 16)
  TEST_LOOP (double, 16)
  TEST_LOOP (float, 77)
  TEST_LOOP (double, 77)
  TEST_LOOP (float, 128)
  TEST_LOOP (double, 128)
  TEST_LOOP (float, 795)
  TEST_LOOP (double, 795)
  return 0;
}