mirror of
https://github.com/OpenMP/Examples.git
synced 2025-04-10 16:32:11 +01:00
25 lines
605 B
C
25 lines
605 B
C
/*
|
|
* @@name: device.2
|
|
* @@type: C
|
|
* @@operation: compile
|
|
* @@expect: success
|
|
* @@version: omp_4.0
|
|
*/
|
|
#include <omp.h>
|
|
extern void init(float *, float *, int);
|
|
extern void output(float *, int);
|
|
void vec_mult(float *p, float *v1, float *v2, int N)
|
|
{
|
|
int i;
|
|
init(v1, v2, N);
|
|
int ndev = omp_get_num_devices();
|
|
int do_offload = (ndev>0 && N>1000000);
|
|
#pragma omp target if(do_offload) \
|
|
map(to: v1[0:N], v2[:N]) \
|
|
map(from: p[0:N])
|
|
#pragma omp parallel for if(N>1000) private(i)
|
|
for (i=0; i<N; i++)
|
|
p[i] = v1[i] * v2[i];
|
|
output(p, N);
|
|
}
|