1 | // RUN: %libomptarget-compileopt-run-and-check-generic |
---|---|
2 | // |
3 | // REQUIRES: gpu |
4 | |
5 | #include <omp.h> |
6 | #include <ompx.h> |
7 | #include <stdio.h> |
8 | #include <stdlib.h> |
9 | #include <string.h> |
10 | |
11 | struct info { |
12 | int tid, bid, tdim; |
13 | }; |
14 | |
15 | int main(int argc, char **argv) { |
16 | int N = 1 << 20; |
17 | if (argc > 1) |
18 | N = atoi(nptr: argv[1]); |
19 | |
20 | struct info *X = (struct info *)malloc(size: sizeof(*X) * N); |
21 | memset(s: X, c: '0', n: sizeof(*X) * N); |
22 | |
23 | int TL = 256; |
24 | int NT = (N + TL - 1) / TL; |
25 | |
26 | #pragma omp target data map(tofrom : X [0:N]) |
27 | #pragma omp target teams num_teams(NT) thread_limit(TL) |
28 | { |
29 | #pragma omp parallel |
30 | { |
31 | int tid = ompx_thread_id_x(); |
32 | int bid = ompx_block_id_x(); |
33 | int tdim = ompx_block_dim_x(); |
34 | int gid = tid + bid * tdim; |
35 | if (gid < N) { |
36 | X[gid].tid = tid; |
37 | X[gid].bid = bid; |
38 | X[gid].tdim = tdim; |
39 | }; |
40 | } |
41 | } |
42 | |
43 | int tid = 0, bid = 0, tdim = 256; |
44 | for (int i = 0; i < N; i++) { |
45 | if (X[i].tid != tid || X[i].bid != bid || X[i].tdim != tdim) { |
46 | printf(format: "%i: %i vs %i, %i vs %i, %i vs %i\n", i, X[i].tid, tid, X[i].bid, |
47 | bid, X[i].tdim, tdim); |
48 | return 1; |
49 | } |
50 | tid++; |
51 | if (tid == tdim) { |
52 | tid = 0; |
53 | bid++; |
54 | } |
55 | } |
56 | |
57 | // CHECK: OK |
58 | printf(format: "OK"); |
59 | return 0; |
60 | } |
61 |