mirror of
https://github.com/OpenMP/Examples.git
synced 2025-04-07 23:22:11 +01:00
44 lines
961 B
C
44 lines
961 B
C
/*
|
|
* @@name: device.1c
|
|
* @@type: C
|
|
* @@compilable: yes
|
|
* @@linkable: no
|
|
* @@expect: success
|
|
*/
|
|
#include <stdio.h>
|
|
#include <omp.h>
|
|
#pragma omp declare target
|
|
void vec_mult(float *p, float *v1, float *v2, int N);
|
|
extern float *p, *v1, *v2;
|
|
extern int N;
|
|
#pragma omp end declare target
|
|
extern void init_vars(float *, float *, int);
|
|
extern void output(float *, int);
|
|
void foo()
|
|
{
|
|
N = init_vars(&p, &v1, &v2);
|
|
#pragma omp target device(42) map(p[:N], v1[:N], v2[:N])
|
|
{
|
|
vec_mult(p, v1, v2, N);
|
|
}
|
|
output(p, N);
|
|
}
|
|
void vec_mult(float *p, float *v1, float *v2, int N)
|
|
{
|
|
int i;
|
|
int nthreads = omp_is_initial_device() ? 8 : 1024;
|
|
if (!omp_is_initial_device())
|
|
{
|
|
printf("1024 threads on target device\n");
|
|
nthreads = 1024;
|
|
}
|
|
else
|
|
{
|
|
printf("8 threads on initial device\n");
|
|
nthreads = 8;
|
|
}
|
|
#pragma omp parallel for private(i) num_threads(nthreads);
|
|
for (i=0; i<N; i++)
|
|
p[i] = v1[i] * v2[i];
|
|
}
|