ubuntu-buildroot/output/build/host-gcc-final-11.4.0/libgomp/testsuite/libgomp.c/examples-4/simd-6.c

106 lines
1.8 KiB
C

/* { dg-do run { target vect_simd_clones } } */
/* { dg-additional-options "-msse2" { target sse2_runtime } } */
/* { dg-additional-options "-mavx" { target avx_runtime } } */
#define N 100
#define EPS 0.000001
#include <stdlib.h>
#include <stdio.h>
void init(int *b, float *y, int n)
{
int i, s = -1;
for ( i=0; i<N; i++ )
{
b[i] = i*i*s;
y[i] = b[i] * 0.1f;
s = -s;
}
}
#pragma omp declare simd linear(p:1) notinbranch
int foo(int *p){
*p = *p + 10;
return *p;
}
int myaddint(int *a, int *b, int n)
{
#pragma omp simd
for (int i=0; i<n; i++){
a[i] = foo(&b[i]); /* foo is not called under a condition */
}
return a[n-1];
}
int myaddint_ref(int *a, int *b, int n)
{
for (int i=0; i<n; i++){
a[i] = foo(&b[i]);
}
return a[n-1];
}
#pragma omp declare simd linear(p:1) inbranch
float goo(float *p){
*p = *p + 18.5f;
return *p;
}
int myaddfloat(float *x, float *y, int n)
{
#pragma omp simd
for (int i=0; i<n; i++){
x[i] = (x[i] > y[i]) ? goo(&y[i]) : y[i];
/* goo is called under the condition (or within a branch) */
}
return x[n-1];
}
int myaddfloat_ref(float *x, float *y, int n)
{
for (int i=0; i<n; i++){
x[i] = (x[i] > y[i]) ? goo(&y[i]) : y[i];
}
return x[n-1];
}
void check_addint (int *a, int *b)
{
int i;
for (i = 0; i < N; i++)
if (a[i] != b[i])
abort ();
}
void check_addfloat (float *a, float *b)
{
int i;
for (i = 0; i < N; i++)
if (a[i] - b[i] > EPS || b[i] - a[i] > EPS)
abort ();
}
int main ()
{
int i;
int a[N], a_ref[N], b[N];
float x[N], x_ref[N], y[N];
init(a, x, N);
init(b, y, N);
myaddint(a, b, N);
myaddfloat(x, y, N);
init(a_ref, x_ref, N);
init(b, y, N);
myaddint_ref(a_ref, b, N);
myaddfloat_ref(x_ref, y, N);
check_addint(a, a_ref);
check_addfloat(x, x_ref);
return 0;
}