mirror of
https://github.com/OpenMP/Examples.git
synced 2025-04-10 16:32:11 +01:00
47 lines
1.1 KiB
C
47 lines
1.1 KiB
C
/*
|
|
* @@name: async_target.2
|
|
* @@type: C
|
|
* @@operation: compile
|
|
* @@expect: success
|
|
* @@version: omp_5.1
|
|
*/
|
|
#include <stdlib.h>
|
|
#include <omp.h>
|
|
|
|
#pragma omp begin declare target
|
|
extern void init(float *, float *, int);
|
|
#pragma omp end declare target
|
|
|
|
extern void foo();
|
|
extern void output(float *, int);
|
|
void vec_mult(float *p, int N, int dev)
|
|
{
|
|
float *v1, *v2;
|
|
int i;
|
|
#pragma omp task shared(v1, v2) depend(out: v1, v2)
|
|
#pragma omp target device(dev) map(v1, v2)
|
|
{
|
|
// check whether on device dev
|
|
if (omp_is_initial_device())
|
|
abort();
|
|
v1 = (float *)malloc(N*sizeof(float));
|
|
v2 = (float *)malloc(N*sizeof(float));
|
|
init(v1, v2, N);
|
|
}
|
|
foo(); // execute other work asychronously
|
|
#pragma omp task shared(v1, v2, p) depend(in: v1, v2)
|
|
#pragma omp target device(dev) map(to: v1, v2) map(from: p[0:N])
|
|
{
|
|
// check whether on device dev
|
|
if (omp_is_initial_device())
|
|
abort();
|
|
#pragma omp parallel for
|
|
for (i=0; i<N; i++)
|
|
p[i] = v1[i] * v2[i];
|
|
free(v1);
|
|
free(v2);
|
|
}
|
|
#pragma omp taskwait
|
|
output(p, N);
|
|
}
|